天天看點

cuda安裝

顯示卡、顯示卡驅動、CUDA之間的關系

顯示卡:(GPU)主流是NVIDIA的GPU,深度學習本身需要大量計算。GPU的并行計算能力,在過去幾年裡恰當地滿足了深度學習的需求。AMD的GPU基本沒有什麼支援,可以不用考慮。

驅動:沒有顯示卡驅動,就不能識别GPU硬體,不能調用其計算資源。

CUDA:是NVIDIA開發的一種并行計算平台和程式設計模型,用于圖形處理單元(GPU)上的正常計算。借助CUDA,開發人員能夠利用GPU的功能來顯着加快計算應用程式的速度。

cuda 部署安裝包下載下傳:https://developer.nvidia.com/cuda-downloads?target_os=Linux

cuda安裝

wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux.run

sh cuda_11.4.2_470.57.02_linux.run

cuda安裝
cuda安裝
cuda安裝

配置環境變量

vim /etc/profile

export PATH="$PATH:/usr/local/cuda-11.4/bin"

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-11.4/lib64"

source /etc/profile

驗證CUDA

nvcc -V