GPU計算力分析
https://blog.csdn.net/iefenghao/article/details/97956440?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.edu_weight&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.edu_weight
Table 2. CUDA Toolkit and Minimum Compatible Driver Versions
CUDA Toolkit | Linux x86_64 Driver Version | Windows x86_64 Driver Version |
---|---|---|
CUDA 11.2.1 Update 1 | >=460.32.03 | >=461.09 |
CUDA 11.2.0 GA | >=460.27.03 | >=460.82 |
CUDA 11.1.1 Update 1 | >=455.32 | >=456.81 |
CUDA 11.1 GA | >=455.23 | >=456.38 |
CUDA 11.0.3 Update 1 | >= 450.51.06 | >= 451.82 |
CUDA 11.0.2 GA | >= 450.51.05 | >= 451.48 |
CUDA 11.0.1 RC | >= 450.36.06 | >= 451.22 |
CUDA 10.2.89 | >= 440.33 | >= 441.22 |
CUDA 10.1 (10.1.105 general release, and updates) | >= 418.39 | >= 418.96 |
CUDA 10.0.130 | >= 410.48 | >= 411.31 |
CUDA 9.2 (9.2.148 Update 1) | >= 396.37 | >= 398.26 |
CUDA 9.2 (9.2.88) | >= 396.26 | >= 397.44 |
CUDA 9.1 (9.1.85) | >= 390.46 | >= 391.29 |
CUDA 9.0 (9.0.76) | >= 384.81 | >= 385.54 |
CUDA 8.0 (8.0.61 GA2) | >= 375.26 | >= 376.51 |
CUDA 8.0 (8.0.44) | >= 367.48 | >= 369.30 |
CUDA 7.5 (7.5.16) | >= 352.31 | >= 353.66 |
CUDA 7.0 (7.0.28) | >= 346.46 | >= 347.62 |
1、NVidia顯示驅動安裝:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
GPU顯示驅動安裝:
在docker終端輸入
ubuntu-drivers devices,檢視顯示卡配置資訊,正常顯示如下:
root@e25881d8e29d:/# ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001F36sv0000103Csd0000860Fbc03sc00i00
vendor : NVIDIA Corporation
driver : nvidia-driver-450-server - distro non-free
driver : nvidia-driver-440-server - distro non-free
driver : nvidia-driver-460 - distro non-free recommended
driver : nvidia-driver-450 - distro non-free
driver : nvidia-driver-418-server - distro non-free
driver : xserver-xorg-video-nouveau - distro free builtin
== /sys/devices/pci0000:00/0000:00:1d.5/0000:70:00.0 ==
modalias : pci:v00008086d00002723sv00008086sd00000088bc02sc80i00
vendor : Intel Corporation
driver : backport-iwlwifi-dkms - distro free
如果提示
bash: ubutnu-drivers: command not found
,則安裝:
sudo apt-get install ubuntu-drivers-common
安裝顯示卡驅動:
sudo ./NVIDIA-Linux-x86_64-460.39.run -no-x-check -no-nouveau-check -no-opengl-files //隻有禁用opengl這樣安裝才不會出現循環登陸的問題 Ubuntu 16.04安裝NVIDIA驅動後導緻的循環登入問題
-no-x-check:安裝驅動時關閉X服務
-no-nouveau-check:安裝驅動時禁用nouveau
-no-opengl-files:隻安裝驅動檔案,不安裝OpenGL檔案
或者
sudo apt-get install nvidia-driver-460 (460.32)
檢視驅動版本:
cat /proc/driver/nvidia/version
root@e25881d8e29d:/workspace# cat /proc/driver/nvidia/version
NVRM version: NVIDIA UNIX x86_64 Kernel Module 450.102.04 Tue Dec 29 06:51:23 UTC 2020
GCC version: gcc version 7.5.0 (Ubuntu 7.5.0-3ubuntu1~18.04)
解除安裝GPU驅動:
sudo apt-get purge nvidia*
sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
sudo apt-get install nvidia-415 nvidia-settings nvidia-prime
nvidia-smi
2、Cuda安裝
ls /dev/nvidia*,顯示如下:
/dev/nvidia-modeset /dev/nvidia-uvm /dev/nvidia-uvm-tools /dev/nvidia0 /dev/nvidiactl
/dev/nvidia-caps:
nvidia-cap1 nvidia-cap2
CUDA版本下載下傳連結:https://developer.download.nvidia.cn/compute/cuda/repos/ubuntu1804/x86_64/
先下載下傳與顯示卡驅動版本對應的CUDA包:
dpkg -i cuda-repo-ubuntu1804-11-2-local_11.2.1-460.32.03-1_amd64.deb,存入目前目錄
#install CUDA
sudo dpkg -i sudo dpkg -i cuda-repo-ubuntu1804-11-2-local_11.2.1-460.32.03-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1804-11-2-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
安裝完之後,需要設定環境變量
sudo gedit ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda/lib
export PATH=$PATH:/usr/local/cuda/bin
然後,輸入以下指令使配置的環境變量立即生效:
source ~/.bashrc
檢視目前cuda是否成功安裝
nvcc -V
root@e25881d8e29d:/usr/local/cuda-10.2/lib64# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89
出現以上資訊,說明cuda10.2即已安裝完成。
3.CUDNN安裝
先下載下傳好包libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb要本地目錄
sudo dpkg -i libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb
碰到一個問題:先安裝好GPU顯示卡,然後再裝CUDA,CUDNN,重新開機下顯示卡驅動直接掉了,需要再次安裝GPU顯示卡,nvidia-smi顯示460.39,一切正常!