显卡 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就按装好了~~
© 版权声明
文章版权归作者所有,未经允许请勿转载。如内容涉嫌侵权,请在本页底部进入<联系我们>进行举报投诉!
THE END














暂无评论内容