首先顯卡驅動
CUDA 選擇配合顯卡驅動的版本,具體可以參考https://blog.csdn.net/qq_27825451/article/details/89082978
CUDA下載地址為 https://developer.nvidia.com/cuda-downloads
下載后,運行
sudo bash cuda_9.1.85_387.26_linux.run
依次輸入continue\accept,最后不要勾選顯卡驅動(因為顯卡驅動已經安裝好了)
之后據彈出的Summary配置環境變量
sudo vim ~/.bashrc
在文件末尾添上以下兩句(其中local/cuda部分,cuda根據安裝版本修改名字
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
然后使環境變量生效
source ~/.bashrc
最后檢查是否裝上
nvcc -V
若顯示以下界面則安裝成功