【环境配置】多cuda版本多pytorch版本使用。
CUDA 11.3+pytorch 1.12.1
1.安装CUDA11.3
2.安装cudnn版本。注意,这里使用安装到CUDA文件夹的方式。
安装cudnn 到cuda-11.3文件夹:
1.下载cudnn的压缩包。在 cudnn下载中选择for 11.x。下载时选择:Local Installer for Linux x86_64 (Tar)。
2.解压:
tar cudnn-10.1-linux-x64-v8.0.4.30.tgz #解压
3.拷贝到cuda文件夹
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-11.3/lib64/libcudnn*
3.安装pytorch 1.12.1。
写入cuda路径到YOUR_USER/bashsrc:
sudo vim ~