一、如果没有驱动,首先安装驱动:
1. 使用cuda自带驱动:
登陆网址选择需要的,安装时选择安装nvidia驱动
2. 自行安装驱动:
(1)(option)如果有之前的,先卸载
通过ubuntu自带软件或者ppa下载的驱动:
sudo apt-get remove --purge nvidia*
sudo apt autoremove
对于通过官网.run文件,卸载指令如下
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-XXX.XX.XX.run --uninstall
(2)然后安装新的
sudo ubuntu-drivers devices(可用驱动,建议选择recommend)
sudo apt install nvidia-driver-XXX
如果出现问题参考ubuntu20.04安装nvidia驱动真实有效(被折磨了一天一夜的肝文!!!)_nvidia-driver-525 : 依赖: libnvidia-gl-525 (= 525.14-CSDN博客
具体就是:
Ctrl+PgDn 两个按钮同时按,弹出一个输入密码的界面
设置密码,需要设置两次,两次都一样
安装完后reboot重启
选择Enroll mok,回车;
选择continue,回车;
选择yes,回车;
最后就会弹出让输入密码的界面,就是输入刚上一步设置的密码。
输入reboot重启,nvidia-smi检查驱动
二、安装CUDA:
1. 安装cuda时同时安装驱动后自带:
在一.1中已说明
2. 在安装完驱动后手动安装CUDA:
CUDA Toolkit Archive | NVIDIA Developer
选择.run版本,根据官网给出的命令执行:
(下载)wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
(安装)sudo sh cuda_11.8.0_520.61.05_linux.run
可能遇到的问题:
gcc版本不对,可能需要将版本改到gcc7
没有权限,使用sudo执行命令
安装完成后需要在系统变量中添加cuda路径
sudo gedit ~/.bashrc
在里面添加:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
export PATH=$PATH:/usr/local/cuda-11.8/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.8保存后退出
source ~/.bashrc
使用 nvcc -V 检查是否安装成功
三、安装cuDNN:
登陆官网下载cnDNN:
注意下载for linux的tar包而不是deb包
下载后到文件终端使用指令将下载的文件复制到cuda文件夹中
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
使用 cat /usr/local/cuda-XX.X/version.json 检查是否安装成功