ubuntu CUDA安装

一、如果没有驱动，首先安装驱动：

1. 使用cuda自带驱动：

登陆网址选择需要的，安装时选择安装nvidia驱动

https://developer.nvidia.com/cuda-toolkit-archive

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：

cuDNN Archive | NVIDIA Developer

注意下载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 检查是否安装成功