安装nvidia驱动
首先使用ppa安装,添加ppa源
sudo add-apt-repository ppa:graphics-drivers/ppa
然后更新一下
sudo apt-get update
如果添加源成功则输入:
ubuntu-drivers devices
则会显示你的nvidia的显卡型号和推荐装的驱动
然后输入:
sudo ubuntu-drivers autoinstall
其自动会帮助你安装最新版本的驱动,以及安装好依赖项
安装完成以后再输入:
sudo ubuntu-drivers autoinstall
最后重启:
sudo reboot
重启过后输入:
nvidia-smi
查看显卡有无安装成功,若出现有框显示则表示显卡安装成功
安装cuda
首先进入网站:CUDA Toolkit 10.1 Update 1 Download
选择deb(network)安装
然后在下载文件的文件夹打开命令窗口(ctrl+alt+T)依次输入:
sudo dpkg -i cuda-repo-ubuntu1804_10.1.168-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pubhttps://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-libraries-10-1
最后一行与网站上不一致的原因是install cuda 会把驱动也一起打包下载了,但是我们已经安装了驱动,所以只需要下载cuda的包就可以了
安装好了之后接着再输入:
sudo gedit ~/.bashrc
在这个文档的最下面添加路径:
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
保存退出接着再输入命令启用环境:
source ~/.bashrc
最后在命令行中输入命令:
nvcc -V
若出现cuda版本,即安装成功!
安装cudnn的deb版
cudnn download
选择deb版安装:
[cuDNN Runtime Library for Ubuntu18.04 (Deb)](https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.6.1.34/prod/10.1_20190620/Ubuntu18_04-x64/libcudnn7_7.6.1.34-1%2Bcuda10.1_amd64.deb)
[cuDNN Developer Library for Ubuntu18.04 (Deb)](https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.6.1.34/prod/10.1_20190620/Ubuntu18_04-x64/libcudnn7-dev_7.6.1.34-1%2Bcuda10.1_amd64.deb)
[cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)](https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.6.1.34/prod/10.1_20190620/Ubuntu18_04-x64/libcudnn7-doc_7.6.1.34-1%2Bcuda10.1_amd64.deb)
依次将其下载下来,通过所在文件夹的命令窗口输入:
sudo dpkg -i <runtime library的包名>
sudo dpkg -i <Develogper library 的包名>
sudo dpkg -i <code Sameple library 的包名>
到此环境就已经安装完毕了,如果想要测试cudnn有没有安装成功
请转此链接文章末尾:验证cudnn是否安装成功