CUDA11.4、CUDNN、Pytorch安装
一、安装CUDA 11.4
- 下载CUDA 11.4
访问NVIDIA官网下载CUDA 11.4安装包。需要注册NVIDIA账号才能下载。 - 安装CUDA 11.4
解压下载的安装包，打开CUDA 11.4文件夹，找到setup.exe，双击运行。
根据提示，选择“Custom”安装类型，选择CUDA安装目录（建议默认路径），CUDA工具包安装目录（建议默认路径），CUDNN安装目录（建议默认路径）。
勾选“Install NVIDIA Accelerated Driver for Compute, Graphics, and猝碰 for Linux”选项，选择“No，not this time”选项。
点击“Next”，进入“Accept the terms in the License Agreement”页面，勾选“Accept the NVIDIA software license”选项。
点击“Next”，进入“Select Components”页面，勾选CUDA运行时组件，根据需要勾选CUDA开发工具包，并选择要安装的组件。
点击“Next”，进入“Select Destination Location”页面，默认选择CUDA安装目录。
点击“Next”，进入“Select Prerequisites”页面，根据需要选择要安装的依赖包。
点击“Next”，进入“Ready to Install”页面，确认安装信息无误后，点击“Install”按钮开始安装。
等待安装完成后，点击“Finish”按钮完成CUDA 11.4的安装。
二、安装CUDNN - 下载CUDNN
访问NVIDIA官网下载CUDNN安装包。需要注册NVIDIA账号才能下载。 - 解压CUDNN安装包
将下载的CUDNN安装包解压到CUDA 11.4文件夹中。 - 配置环境变量
在终端中执行以下命令，将CUDNN库文件路径添加到LD_LIBRARY_PATH环境变量中：
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:$HOME/cuda/lib64:$HOME/cuda/extras/CUPTI/lib64:$HOME/cuda/extras/visualization_tools/lib64:$HOME/cuda/extras/visualization_samples:$HOME/cuda/extras/visualization_apps:$HOME/cuda/extras/nsight_compute:$HOME/cuda/extras/nsight_viz:$HOME/cuda/extras/nsight_perfapi:$HOME/cuda/extras/nsight_sysmon:$HOME/cuda/extras/nsight_debug:$HOME/cuda/extras/nsight_battery:$HOME/cuda/extras/nsight_charmony:$HOME/cuda/extras/nsight_trappers:$HOME/cuda/extras/nsight_diamond:$HOME/cuda/extras/nsight_optimus:$HOME/cuda/extras/nsight_all_apps:$HOME/cuda/extras/nsight_compute:$HOME/cuda/extras/$LOWERCASE_MODE