本文是在Ubuntu18.04.5伺服器上安裝CUDA_10.1(nvidia-driver455)和cuDNN_7.6.5,
- Ubuntu 18.04.5
- CUDA_10.1 (nvidia-driver455)
- cuDNN_7.6.5
一、 前期準備
1、檢視系統版本和顯示卡型号cat
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLicGcq5iYfJ2YkFGMyYWO1UDMzAjZ5UGO1cDOjZGZ3YTO4UjMwUWMtIjdvwVbvNmLn1WaopnL0MWaw9CXvwlOzBHd0hWPsJXdmYDM3YjZkJGNzQDNl1SOhRGOtETMiVWLhJTOz0iZ4QmM2ETYl1DZpV3ZmITPlBXe0ZyPldWYtl2LcdXZpZ3Lc12bj5SZjVjL5h3byBnLzATLn1Wavw1LcpDc0RHaiojIsJye.jpg)
nvidia
3、顯示卡支援的驅動型号 ubuntu
此處可以看到系統推薦的驅動版本是 455:
4、安裝推薦版本455的顯示卡驅動sudo
5、重新開機伺服器(無其它使用者使用伺服器) (
6、檢視顯示卡驅動是否安裝成功: nvidia
安裝成功會顯示
$
二、 RUN方式安裝CUDA _10.1 (我有安裝過11.0,11.1,後因項目需求又改為10.1,圖檔有點混)
0、安裝之前需要确認已安裝gcc和g++$
1、[官網cuda_10.1 下載下傳位址]( CUDA Toolkit 10.1 original Archive ): 2、點選下載下傳:cuda_10.1.105_418.39_linux.run 上傳
cuda_10.1.105_418.39_linux.run檔案至
ubuntu系統上的
home/your_username目錄下
4、執行指令安裝sudo
5、同意協定:輸入accept 最後選擇
install- ~.bashrc # 修改home目錄下的.bashrc檔案,隻針對目前使用者
- /etc/profile # 修改profile檔案,針對所有使用者
# 修改home目錄下的.bashrc檔案
# 修改profile檔案,針對所有使用者
9、立即生效(或重新開機伺服器生效) source .bashrc
或
source /etc/profile
10、驗證安裝成功 nvcc -V (注意 是大V) nvcc -V
cd /usr/local/cuda-11.0/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0
三、 deb方式安裝cuDNN_7.6.5
1、[cudnn官方下載下傳位址](cuDNN Archive)- tar方式: 直接下載下傳 cuDNN Library for Linux
- deb方式: 需要下載下傳 runtime 、developer 和 Samples
推薦使用deb方式安裝:使用tar方式安裝沒有cudnn_doc_v7檔案,無法驗證是否安裝成功;
3、将下載下傳的三個檔案copy至 home/your_username目錄下sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb
5、測試cuDNN是否安裝成功 1)複制cuDNN samples到home目錄下
$ cp -r /usr/src/cudnn_samples_v7 /$HOME
2) 進入home目錄
$ cd $HOME/cudnn_samples_v7/mnistCUDNN/
3) 編譯mnistCUDNN
$ sudo make clean
$ sudo make
4)運作mnistCUDNN
$ sudo ./mnistCUDNN
(運作需要一段時間)如果出現Test passed! 表明cuDNN已安裝成功。
over! 已成功安裝nvidia-driver455 、cuda10.1 、cuDNN_7.6.5。
4、本文參考文章:
[ 1)、Linux 下的 CUDA 安裝和使用指南](「已登出」:Linux 下的 CUDA 安裝和使用指南)
[2)、NVIDIA cuDNN v8 deb方法安裝教程](李森科在zhihu:NVIDIA cuDNN v8 deb方法安裝教程(Linux/Ubuntu))
[3)、Linux下安裝CUDA并配置環境變量](Linux下安裝CUDA并配置環境變量_liu_feng_zi_的專欄-CSDN部落格_linux配置cuda環境變量)
[4)、Linux檢視GPU資訊和使用情況](Linux檢視GPU資訊和使用情況 - FerdyCorp - 部落格園)