一、安装nvidia-driver

运行 ubuntu-drivers devices

(base) diraw@solitude:~$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00002520sv00001043sd00001CBCbc03sc00i00
vendor   : NVIDIA Corporation
model    : GA106M [GeForce RTX 3060 Mobile / Max-Q]
driver   : nvidia-driver-570-server-open - distro non-free
driver   : nvidia-driver-550-open - distro non-free
driver   : nvidia-driver-545 - distro non-free
driver   : nvidia-driver-470-server - distro non-free
driver   : nvidia-driver-570-server - distro non-free
driver   : nvidia-driver-535-open - distro non-free
driver   : nvidia-driver-470 - distro non-free
driver   : nvidia-driver-535 - distro non-free
driver   : nvidia-driver-550 - distro non-free recommended
driver   : nvidia-driver-545-open - distro non-free
driver   : nvidia-driver-535-server - distro non-free
driver   : nvidia-driver-535-server-open - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

其中550版本被标记为recommended（实测）

运行sudo apt install nvidia-driver-550

安装错了的话，使用apt卸载

sudo apt purge nvidia-*
sudo apt autoremove

之后会让你输入密码，完成之后sudo reboot，出现下图：

Image 6

选择第二行Enroll MOK，然后出现下图：

Image 7

选择continue

Image 8

然后选择reboot

重启之后，运行nvidia-smi，预期出现下图，右上角为所支持的最大版本cuda，我这个为12.04

Image 9

二、安装CUDA Toolkit

去 https://developer.nvidia.com/cuda-toolkit-archive ，选择12.04

Image 10

然后选择你的配置

Image 11

然后把下面的代码走一遍

Image 12

传统内核模块:可能与旧系统或需要传统驱动结构的特定软件具有更好的兼容性。

开放内核模块:使用开源内核模块，可以更好地与 Linux 内核集成，潜在地提高安全性并加快更新速度。

这里选择第二个，以及如果第一步操作了，nvidia-driver已经安装了，执行第二条即可

Image 13

通过sudo apt install nvidia-cuda-toolkit安装

检查成功：nvcc --version

三、添加环境变量

nano ~/.bashrc

添加以下行到文件末尾

export CUDA_HOME=/usr/local/cuda export PATH=$CUDA_HOME/bin:$PATH export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

应用更改source ~/.bashrc