ubuntu 安装 cuda 驱动

ubuntu 安装 cuda 驱动

1 下载 cuda driver

下载地址：https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local

wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
sudo sh cuda_12.1.0_530.30.02_linux.run

问题

如果出现报错 failed
参考博客
https://blog.csdn.net/bocai1215/article/details/126559197?spm=1001.2101.3001.6661.1&utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-126559197-blog-125124730.pc_relevant_3mothn_strategy_and_data_recovery&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-126559197-blog-125124730.pc_relevant_3mothn_strategy_and_data_recovery&utm_relevant_index=1

检查是否安装成功

peter@peter-B760M-GAMING-AC-DDR4:~$ nvidia-smi
Thu Mar 16 17:05:20 2023       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 530.30.02              Driver Version: 530.30.02    CUDA Version: 12.1     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                  Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf            Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 3060 Ti      Off| 00000000:01:00.0 Off |                  N/A |
|  0%   34C    P8               12W / 265W|    213MiB /  8192MiB |      1%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|    0   N/A  N/A       893      G   /usr/lib/xorg/Xorg                           53MiB |
|    0   N/A  N/A      1473      G   /usr/lib/xorg/Xorg                           62MiB |
|    0   N/A  N/A      1599      G   /usr/bin/gnome-shell                         86MiB |
+---------------------------------------------------------------------------------------+