显卡 NVIDIA RTX 3060只能支持CUDA11以上版本,只能卸载 CUDA 10.2,再安装CUDA 11.3
环境:Ubuntu 16.04
具体参考:https://blog.csdn.net/qq_21570025/article/details/115615664 (如果侵权,速删)
1.卸载CUDA 10.2
cd /usr/local
sudo rm -rf cuda
sudo rm -rf cuda-10.2
删除 ~/.bashrc文件内的内容
vim ~/.bashrc
删除后source一下
source ~/.bashrc
2.安装CUDA 11.3
官网地址:https://developer.nvidia.com/cuda-11.3.0-download-archive
选择Linux - x86_64 - Ubuntu - 16.04 - runfile(local)
执行官网命令:
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run
选择accept
如果安装了device,记得去掉,并且选择install
查看CUDA版本号:
nvcc -V
如果提示:程序“nvcc”尚未安装。 您可以使用以下命令安装:sudo apt install nvidia-cuda-toolkit
千万别用这个命令!等待时间太长了!
直接执行命令:
vim ~/.bashrc
//在文件中加入三行
export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=usr/local/cuda-11.3$CUDA_HOME
别忘source一下
source ~/.bashrc
//查看版本号
nvcc -V
CUDA 11.3就安装好了~
3.安装cuDNN
官网地址:https://developer.nvidia.com/rdp/cudnn-archive (按需自取)
下载CUDA 11.3对应的 cuDNN 8.2.0
选择 cuDNN Library for Linux [x86_64]
进入下载的文件夹
解压至当前文件夹
tar -xzvf cudnn-11.3-linux-x64-v8.2.0.53.tgz
文件夹内会出现一个cuda的文件夹
运行如下命令:
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
查看版本号
注意:cuDNN8.0之后的新版本将版本号放到cudnn_version.h中
cat /usr/local/cuda-11.3/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
如果失败的话,重启电脑试试。
这样cuDNN8.2.0就按装好了~~