1.禁用自带的 nouveau nvidia驱动
sudo vim /etc/modprobe.d/blacklist.conf
在文件夹最末尾处添加命令
blacklist nouveauoptions nouveau modeset=0
2.更新内核
sudo update-initramfs -u
3.重启系统
sudo reboot
4.在登录界面按Ctrl + Alt + F1键(如果是ssh直接查看下面命令输出)
lsmod | grep nouveau
查看nouveau模块是否被加载,如果什么都没有输出,则执行下一步,不行的话重新操作1-3
5.下载显卡驱动
在英伟达驱动下载页面上选择对应的系统和显卡型号,下载驱动
/Download/index.aspx
Titan RTX的440.44驱动链接
http://us./XFree86/Linux-x86_64/440.44/NVIDIA-Linux-x86_64-440.44.run
直接下载到服务器上
wget http://us./XFree86/Linux-x86_64/440.44/NVIDIA-Linux-x86_64-440.44.run
cd到你下载的显卡驱动的路径
sudo chmod 777 NVIDIA-Linux-x86_64-440.44.run(驱动文件名) #对驱动文件赋权sudo service lightdm stop #关闭集显
6.安装驱动
sudo bash NVIDIA-Linux-x86_64-440.44.run
按照提示安装
安装完成后可以使用nvidia-smi命令测试一下
出现上面的显卡信息就说明驱动已经安装成功了
7.重启
重启一下,准备安装cuda和cudnn
安装cuda-10.0
从官网上下载cuda-10.0的.run文件
CUDA Toolkit
安装
sudo bash cuda_10.0.130_410.48_linux.run
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?(y)es/(n)o/(q)uit: no不安装cuda带的驱动Install the CUDA 10.0 Toolkit?(y)es/(n)o/(q)uit: yesEnter Toolkit Location[ default is /usr/local/cuda-10.0 ]: Do you want to install a symbolic link at /usr/local/cuda?(y)es/(n)o/(q)uit: yInstall the CUDA 10.0 Samples?(y)es/(n)o/(q)uit: yEnter CUDA Samples Location[ default is /home/user ]:
设置环境变量
针对当前用户
打开主目录下的 .bashrc文件添加如下路径,vim ~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64export PATH=$PATH:/usr/local/cuda-10.0/binexport CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
运行source ~/.bashrc
针对所有用户
打开/etc/profile文件添加如下路径sudo vim /etc/profile
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64export PATH=$PATH:/usr/local/cuda-10.0/binexport CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
验证cuda安装
nvcc -V
有上面这样的输出说明安装成功
安装CUDNN
在nvidia官网上下载cudnn的deb包
CUDNN-link
下载
cuDNN Runtime Library for Ubuntu16.04 (Deb)
cuDNN Developer Library for Ubuntu16.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)
这三个文件
依次安装
sudo dpkg -i libcudnn7_7.6.2.24-1+cuda10.0_amd64.debsudo dpkg -i libcudnn7-dev_7.6.2.24-1+cuda10.0_amd64.debsudo dpkg -i libcudnn7-doc_7.6.2.24-1+cuda10.0_amd64.deb
验证cudnn安装
上面安装的第三个包中有官方给的示例,可以用来验证cudnn是否安装成功,示例代码保存在/usr/src/cudnn_samples_v7/中
cp -r /usr/src/cudnn_samples_v7 ~/cudnn-samplecd ~/cudnn-sample/mnistCUDNN/make clean && make -j16./mnistCUDNN
如果运行./mnistCUDNN最后输出test passed,就代表cudnn安装成功