本篇文章是在你的台式電腦已經安裝好NVIDIA驅動的前提下繼續安裝CUDA10.2的教程
一、前期准備
1、首先確定適合自己驅動的CUDA版本,打開命令行輸入命令
nvidia-smi
可以看到的版本(CUDA Version)我的是10.2
2、去官網上找到自己對應的版本(runfile文件)
https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal
3、按照要求安裝cuda
復制下載鏈接,命令行切換到你想要跳轉的位置,我為了方便下載到命令行一打開的位置
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
至此前期准備結束
二、安裝CUDA
1、ctrl+alt+F1 進入命令行界面,然后圖形界面
sudo service lightdm stop
2、進入到之前下載的cuda安裝文件的位置,執行安裝命令
sudo sh cuda_10.2.89_440.33.01_linux.run
3、安裝過程中注意事項
大致注意以下幾點就好:
a)不在安裝NVIDIA顯卡驅動
b)安裝路徑默認
c)同意創建軟連接
d)不用安裝測試(samples)
最后選擇Install,按enter開始安裝。
安裝完成之后最好記下安裝路徑,方便后面配置環境變量
4、重新打開圖形界面
sudo service lightdm restart
5、最好重啟一下計算機
sudo reboot
三、配置環境變量
1、打開文件
sudo gedit ~/.bashrc
2、末尾加入
export PATH=/usr/local/cuda-10.2/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
注意兩個地址和他給出的一樣哦!!!
3、更新文件
source ~/.bashrc
四、驗證安裝
1、重啟計算機或者重新打開一個命令行
2、查看CUDA版本
nvcc -V
如果出現版本信息安裝基本沒有問題了
*** 福利 ***
CUDA卸載
在命令行中輸入
1
2
3
|
sudo apt-get remove cuda
sudo apt-get autoclean
sudo apt-get remove cuda*
|
然后在目錄切換到usr/local/下
1
2
|
cd /usr/local/
sudo rm -r cuda-10.2
|
cuDNN安裝
據說CUDA 10.2 和 cuDNN 7.6.5 更配哦!親!
下載對應版本cuDNN https://developer.nvidia.com/cudnn
1
2
3
4
5
|
tar xvzf cudnn-10.2-linux-x64-v7.6.5.32.tag
sudo cp -P 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*
sudo ldconfig
|
驗證cuDNN安裝
在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
出現以下信息則表明安裝成功
—————— 福利區結束啦 ——————
biu~~~biu~~~