安装NVIDIA驱动
打开:系统设置-软件和更新-附加驱动,选择合适的NVIDIA驱动,点击应用更改,等待读条完成。较为便捷。
但可能有部分电脑无法这样安装,可参考 Ubuntu安装nvidia驱动_小潇港哥的博客-CSDN博客_ubuntu安装nvidia驱动
打开终端,输入
nvidia-smi
这里CUDA Version 11.4 不是指你的电脑已经安装了,而是在你使用的驱动下安装的CUDA版本不能高于显示的版本。
https://developer.nvidia.com/cuda-toolkit-archive
去英伟达找对应的版本
复制Base installer下的两行代码到终端运行
下载指令:
安装指令:
选择continue
输入accept
将光标移动到driver处回车取消安装
再移动光标到install回车
检查是否安装了gcc
export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib64
if存在老版本cuda,先卸载,没有安装过cuda的可以跳过此步(一般系统默认是未装的)
sudo apt-get remove --purge nvidia*
#或者尝试
cuda-uninstaller #卸载cuda,位置在/usr/local/cuda-xx.x/bin。xx为之前安装的版本
nvidia-uninstaller #卸载驱动
禁用nouveau(比较稳妥的安装方法)尽量这样做避免电脑死机
cd /etc/modprobe.d
sudo touch blacklist-nouveau.conf
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
# 在文件末尾添加
blacklist nouveau
options nouveau modeset=0
#保存文件,然后在终端输入
sudo update-initramfs -u
再输入source ~/.bashrc,使设置生效
source ~/.bashrc
检查是否安装成功
nvcc -V
出现下面画面则安装成功
安装 cudnn
进入官网找与cuda版本对应的cudnn
下载 library for linux以及对应版本的runtime, developer,code samples
下载得到这四个文件
在文件下载的路径下打开终端执行下列指令进行安装
tar -zxvf cudnn-11.0-linux-x64-v8.0.4.30.tgz
sudo cp -rf cuda/include/cudnn* /usr/local/cuda/include/
sudo cp -rf cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
sudo dpkg -i libcudnn8_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.4.30-1+cuda11.0_amd64.deb
安装完成后直接在当前终端复制列程到home目录下
cp -r /usr/src/cudnn_samples_v8/ $HOME
转到minist例程
cd $HOME/cudnn_samples_v8/mnistCUDNN
转到minist例程
cd $HOME/cudnn_samples_v8/mnistCUDNN
运行例程
./mnistCUDNN
如果cudnn被恰当的安装,终端最后一行将输出打印消息:Test passed!至此安装成功!