前言
这几天一直在搭建ubuntu16.04下的深度学习环境,遇到许多坑,我将网上不同教程的优点总结了一下,在这里与大家分享。我使用的显卡是RTX2080,CUDA 9.0和CUDNN7_7.4.1.5。CUDA版本需要与CUDNN相适应,具体对应版本在官网下载时能看到。
安装显卡驱动
先到英伟达官网上下载与你的显卡对应的显卡驱动。下载链接:https://www.nvidia.cn/Download/index.aspx?lang=cn 。
下载之后是某一个.run的文件,不同显卡驱动版本可能不一样。
NVIDIA-Linux-x86_64-410.93.run
查看是否有NVIDIA的显卡
lspci | grep -i nvidia
uname -m && cat /etc/*release
有显卡信息代表显卡已经插好。
如果之前有旧的显卡驱动需要先卸载。
(1)进入文本模式:CTRL+ALT+F1
sudo apt-get remove nvidia-*
sudo apt-get autoremove
(2)Uninstall the drivers from the .run file
sudo nvidia-