Ubuntu16.04安裝CUDA10.2+cuDNN7.6.5(福利區),教你如何優雅的安裝環境(1-2)


 

本篇文章是在你的台式電腦已經安裝好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~~~


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM