CUDA(Compute Unified Device Architecture),是顯示卡廠商NVIDIA推出的運算平台。 CUDA™是一種由NVIDIA推出的通用并行計算架構,該架構使GPU能夠解決複雜的計算問題。 它包含了CUDA指令集架構(ISA)以及GPU内部的并行計算引擎。
1. 下載下傳: https://developer.nvidia.com/cuda-downloads
2. 安裝: sh cuda_10.0.130_410.48_linux.sh
Do you accept the previously read EULA?
accept/decline/quit: accept
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit:n
Do you want to install the OpenGL libraries?
(y)es/(n)o/(q)uit [ default is yes ]: n
Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: y
3. 設定環境變量
vim ~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
source ~/.bashrc