1、Ubuntu18.04安装显卡驱动
-
禁用SecureBoot(按照主板型号,查询开机进入bios按键,关闭secureboot)
-
禁用nouveau
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf #在文件中插入以下内容,将nouveau加入黑名单,默认不开启 blacklist nouveau options nouveau modeset=0 #输入以下命令使禁用生效然后重启 sudo update-initramfs -u sudo reboot #重启后验证(输入后没反应即为成功) lsmod | grep nouveau
-
安装驱动
#1、安装前保证驱动卸载干净 sudo ./NVIDIA-Linux-x86_64-384.59.run(修改各自版本号) --uninstall sudo apt-get remove --purge nvidia* #2、ppa源驱动安装,首先查看电脑最适合的版本 ubuntu-drivers devices #加入ppa源 sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo apt-get install nvidia-driver-435 #此处数字要对应上面查询到的版本号 sudo apt-get install mesa-common-dev #安装后重启即可,查看是否安装成功 nvidia-smi
参考:Ubuntu18.04安装Nvidia显卡驱动教程_知遥君的博客-CSDN博客_ubuntu18.04安装nvidia显卡驱动
2、安装cuda
-
首先进入官网下载对应的cuda版本(注:下载cuda前,要搞清楚显卡的型号,深度学习代码要求的环境,如A6000显卡,使用pytorch只能安装cuda11.1版本以上的,否则运行代码时会提示算力不匹配)
-
选择runfile(local)方式下载安装,(速度快)
-
#(注:如显卡驱动已经安装,要取消重复安装显卡,也可以自动配置需要安装的路径,是否覆盖cuda软链接等) sudo sh cuda_10.1.243_418.87.00_linux.run
-
添加环境变量
#在home目录的.bashrc中导入cuda的环境变量 export CUDA_HOME=/usr/local/cuda export PATH=$PATH:$CUDA_HOME/bin export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} #退出.bachrc文件,刷新 source ~/.bashrc
-
输入nvcc -V查看是否安装成功
3、安装cudnn
-
按照官网确定版本号(如A6000在cudnn8是能加快渲染速度)
-
下载,解压
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*
4、最后
如有任何问题,欢迎大家评论留言