ubuntu显卡(NVIDIA)驱动以及对应版本cuda&cudnn安装

(已禁用集显,禁用方法可自行百度)


驱动在线安装方式
进入tty文本模式
ctrl+alt+F1
关闭显示服务
sudo service lightdm stop
卸载原有驱动
sudo apt-get remove --purge nvidia* # 提示有残留可以接 sudo apt autoremove
添加nvidia驱动源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
寻找适合的驱动版本
ubuntu-drivers devices
安装nvidia driver
sudo apt-get install nvidia-430
检查是否安装成功
nvidia-smi
安装成功启动显示服务
sudo service lightdm start
退出tty模式进入图形界面
ctrl+alt+F7


驱动离线安装方式
nvidia官网下载显卡对应的驱动NVIDIA-Linux-x86_64-xxx.xx.run,拷贝到Linux某个目录后先改权限
chomod 777 NVIDIA-Linux-x86_64-xxx.xx.run
卸载原有驱动
sudo apt-get remove --purge nvidia* # 提示有残留可以接 sudo apt autoremove
临时关闭显示服务
sudo service lightdm stop
运行安装程序
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run
安装后再重启显示
sudo service lightdm start


对应版本cuda安装
查看显卡驱动版本对应cuda版本
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
不同版本下载(选择runfile类型的安装文件)
https://developer.nvidia.com/cuda-toolkit-archive
安装
除Install NVIDIA Accelerated Graphics Driver外全部yes或默认
设置环境变量
vim ~/.bashrc #修改配置文件(如果你用的是zsh,则需要修改 ~/.zshrc文件)
#在文件结尾处添加
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
source ~/.bashrc
查看cuda安装版本
nvcc -V
测试cuda与显卡驱动关联成功
cd /usr/local/cuda/samples
sudo make -j52
cd bin/x86_64/linux/release/
./deviceQuery


cuda多个版本的切换
查看安装多个cuda版本
cd /usr/local
ls
查看当前cuda软连接指向版本
stat cuda
切换版本
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda


cudnn的安装
下载安装文件
按需求下载cudnn的安装文件
https://developer.nvidia.com/rdp/cudnn-archive
选择cuDNN Library for Linux
解压cuDNN压缩包
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
复制下列文件到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 /usr/local/cuda/lib64/libcudnn*
查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

原文地址:https://www.cnblogs.com/gawain-ma/p/10860475.html