一、安装显卡驱动
1.更新本地软件
sudo apt-get update
2.查看系统推荐的显卡驱动
ubuntu-drivers devices
3.发现nvidia-driver-470是推荐的显卡驱动,所以输入以下命令
sudo apt-get install nvidia-driver-470
根据系统推荐灵活改变安装的版本
4.重启电脑,输入以下命令,如图所示则代表安装成功
nvidia-smi
二、安装cuda10.2,其他版本同理
1.进入cuda下载网址,找到10.2版本并如图选择信息
2.进入终端,分别输入
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run
出现如下界面,选择continue
随后输入accept
最后一步,由于前面已经安装了驱动,要把安装驱动的选项取消掉,即没有叉号,如图所示,点击回车即可取消 ,最后选择install
3.配置环境变量
在终端中输入sudo gedit ~/.bashrc,将下列代码添加在文件底部,保存并退出。
export PATH="/usr/local/cuda-10.2/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"
最后更新一下环境变量,使软连接生效
sudo source ~/.bashrc
4.验证cuda是否安装成功
nvcc -V
若如下图所示,则代表安装成功
三、安装cudnn8.0.5
1.进入cudnn官网,选择cudnn8.0.5,没有账号须注册才能下载
选择第一个进行下载
下载后,文件出现在下载里,我在下载解压的时候出现无法提取到此处,把文件名改为
cudnn-10.2-linux-x64-v8.0.5.39.tgz
将改好名字的压缩包复制到主目录下,输入以下命令进行解压,会得到一个cuda的文件夹
tar -zxvf cudnn-10.2-linux-x64-v8.0.5.39.tgz
2.复制cuDNN内容到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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
3.检查是否安装成功,输入以下代码,结果如图所示,则代表安装成功
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2