NVIDIA 站点参考:
1. CUDA 12.3 Update 1 Release Notes — Release Notes 12.3 documentation
ubuntu 系统版本:
?lsb_release -a?
python 验证:
import torch
print(torch.version.cuda)
查看相关
nvidia-smi
ubuntu-drivers devices
查看显卡驱动版本:
cat /proc/driver/nvidia/version
查看显卡型号:
lspci | grep -i nvidia
lshw -c video
安装相关
cuda 安装
https://developer.nvidia.com/cuda-toolkit-archive
wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda-repo-ubuntu2204-12-2-local_12.2.0-535.54.03-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-2-local_12.2.0-535.54.03-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-getupdate
?cuDNN安装
https://developer.nvidia.com/rdp/cudnn-download
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb
会提示要安装公钥
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.7.29/cudnn-local-08A7D361-keyring.gpg /usr/share/keyrings/
然后,继续安装
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb
sudo apt-getupdate
卸载NVIDIA Driver
#---open a terminal---
sudo apt-get remove nvidia*
sudo apt autoremove
sudo apt-get install dkms build-essential linux-headers-genericsudo vim /etc/modprobe.d/blacklist.conf
#---save the following info into file blacklist.conf---
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
#---end of the info saved----#---go back to the terminal---
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
sudo update-initramfs -u
reboot
-----------------------------------
参考原文,按上面的步骤一步一步做,完全通过验证
Ubuntu18.04查看显卡信息并安装NVDIA显卡驱动driver + Cuda + Cudnn
https://blog.51cto.com/SpaceVision/4942068
然后安装:
? An alternate method of installing the NVIDIA driver was detected. (This is usually a package provided by your distributor.) A driver installed via that method may integrate better with your system than a driver ? ??
? installed by nvidia-installer.? Please review the message provided by the maintainer of this alternate installation method and decide how to proceed:
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?Continue installation ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?Abort installation ? ?
?The NVIDIA driver provided by Ubuntu can be installed by launching the "Software & Updates" application, and by selecting the NVIDIA driver from the "Additional Drivers" tab.
----》Continue installation ? ? ? ?
Install NVIDIA's 32-bit compatibility libraries?
? ? ? ? ? ? ? ? ? ? ? ? ? Yes ? ? ? ? ? ? ? ? ? ? ? No ??
----》?No
?
? Would you like to register the kernel module sources with DKMS? This will allow DKMS to automatically build a new module, if your kernel changes later.? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? Yes ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? No ??
----->No
? Installation of the NVIDIA Accelerated Graphics Driver for Linux-x86_64 (version: 535.129.03) is now complete.
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? OK ?
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ??
检查驱动是否安装成功:
nvidia-smi
输出如下:
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.129.03 ? ? ? ? ? ? Driver Version: 535.129.03 ? CUDA Version: 12.2 ? ? |
|-----------------------------------------+----------------------+----------------------+
| 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 ?Tesla V100S-PCIE-32GB ? ? ? ? ?Off | 00000000:00:06.0 Off | ? ? ? ? ? ? ? ? ? ?0 |
| N/A ? 29C ? ?P0 ? ? ? ? ? ? ?36W / 250W | ? ? ?0MiB / 32768MiB | ? ? ?0% ? ? ?Default |
| ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? | ? ? ? ? ? ? ? ? ? ? ?| ? ? ? ? ? ? ? ? ?N/A |
+-----------------------------------------+----------------------+----------------------+
| ? 1 ?Tesla V100S-PCIE-32GB ? ? ? ? ?Off | 00000000:00:07.0 Off | ? ? ? ? ? ? ? ? ? ?0 |
| N/A ? 29C ? ?P0 ? ? ? ? ? ? ?37W / 250W | ? ? ?0MiB / 32768MiB | ? ? ?0% ? ? ?Default |
| ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? | ? ? ? ? ? ? ? ? ? ? ?| ? ? ? ? ? ? ? ? ?N/A |
+-----------------------------------------+----------------------+----------------------+
sudo apt install nvidia-driver-535