虽然网上有许多教程,本帖记录下,方便自己以后查找,同时也方便他人。
安装显卡驱动
第一步在英伟达官网下载相对应的显卡驱动
https://www.geforce.cn/drivers
第二步开始安装
禁用nouveau
sudo vim /etc/modprobe.d/blacklist.conf
进入最后一行 添加blacklist nouveau
更新内核
sudo update-initramfs -u
关图形界面
sudo systemctl set-default multiuser.target
重启 sudo reboot
设置显卡驱动权限
sudo chmod 777 ./Nvidia....
(对应自己下载的显卡驱动)
安装显卡驱动
sudo ./Nvidia…..
(同上)
开图形界面
sudo systemctl set-default graphical.target
重启 sudo reboot
开机进入查看显卡驱动是否安装好
nvidia-smi
安装cuda
如果是需要卸载以前的cuda,例如卸载cuda8安装cuda10
卸载cuda
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
删除残余
sudo rm -r /usr/local/cuda-8.0
删除其的版本也是一样的。
安装cuda
下载.run格式的安装说明傻瓜式安装
环境配置
sudo vim ~/.bashrc
在最后面加上
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1
激活
source ~/.bashrc
查看
nvcc -V
cuDnn安装
删除原来的
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
解压下载的
tar -zxvf cudnn-10.1-linux-x64-v7.6.0.64.tgz
将cuda文件夹内的内容分别对应赋值到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*
查看cudnn版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
多个版本的cuda可以并存只要配置好环境变量。