一磁玉、安裝nvidia顯卡驅(qū)動
查看推薦安裝的驅(qū)動
ubuntu-drivers devices
找到推薦的驅(qū)動版本
Selection_002.png
安裝驅(qū)動
sudo apt install nvidia-driver-460
二停忿、安裝cuda
下載對應(yīng)cuda toolkit
https://developer.nvidia.com/cuda-downloads
選擇安裝版本
Selection_004.png
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
sudo sh cuda_11.4.0_470.42.01_linux.run
安裝過程中跳過驅(qū)動安裝
三、cudnn
下載https://developer.nvidia.com/rdp/cudnn-download
下載完成后執(zhí)行
tar -xzvf 剛才下載cudnn壓縮包名稱
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
查看cudnn版本
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
