经测试,Ubuntu20 使用软件和更新安装的nvidia驱动,经过查找,发现并没有带cuda驱动,还是需要手动安装。
参考博客:Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程(图文)
本人决定安装11.3,考虑和另一台电脑版本一致。
网址为:
https://developer.nvidia.com/cuda-toolkit-archive,点击CUDA Toolkit 11.3.1下载相应版本的CUDA11.3.1
在安装之前,需要先安装依赖:
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
往下拉发现安装指令:
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
sudo sh cuda_11.3.1_465.19.01_linux.run
安装过程如下:
- 选择continue
- 输入accept
- 这一步务必把Driver这个选项去掉,因为我们已经自己安装过Nvidia驱动了
- 安装完成后界面如图所示,注意留意下塞纳安装路径,它还给出了卸载cuda的方法。5.CUDA安装完成后,需要配置变量环境才能正常使用。首先在终端输入sudo gedit ~/.bashrc打开如下图所示的.bashrc文件。
然后,如下图所示在.bashrc文件的最后添加以下CUDA环境变量配置信息(我从不同的文章中看到这里添加的信息不仅相同,目前还不太清楚具体含义,所以这里仅仅罗列出它们):
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
或者
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
- 测试是否安装成功
nvcc -V
7. 要是卸载的话,参见ubuntu完全卸载cuda(备忘)
cd /usr/local/cuda-xx.x/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-xx.x
把安装时候装的都卸载。
安装cudnn
下载网址:cuDNN Download(https://developer.nvidia.com/rdp/cudnn-download)。下载Tar格式文件。安装指南网址cudnn安装指南。里面给出了很多种安装方式。
用压缩包安装方式为:
意思就是把名称换一下然后执行指令。简单说就是把cudnn文件夹下include和lib文件夹下的文件分别复制到cuda真实安装路径下对应的include和lib64文件夹下。自己根据实际情况修改。最后一行是赋予读权限。执行时注意替换为自己的真实路径。
$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
查看是否成功(注意路径要是真实安装路径):
cat /usr/local/cuda-11.3/include/cudnn_version.h | grep CUDNN_MAJOR -A 2