NVIDIA顯卡驅(qū)動及CUDA相關(guān)安裝流程

NVIDIA驅(qū)動安裝

NVIDIA驅(qū)動官方下載地址：

https://www.nvidia.cn/Download/index.aspx?lang=cn

第一步先查看適合的顯卡驅(qū)動版本

ubuntu-drivers devices

第二步下載合適的顯卡驅(qū)動

第三步下載驅(qū)動相關(guān)依賴

? 安裝lightdm是在彈窗上選擇lightdm[若沒有彈窗略過]

第四步禁用nouveau驅(qū)動

第五步禁用nouveau內(nèi)核模塊

第六步卸載舊的nvidia驅(qū)動【如果沒裝過，可以省略】

sudo apt-get --purge remove nvidia-*

第七步重啟

sudo reboot

第八步禁用圖形化界面

sudo service lightdm stop

第九步修改文件權(quán)限并安裝

# 先進(jìn)到NVIDIA驅(qū)動所在的文件夾
# 賦予驅(qū)動權(quán)限
  sudo chmod a+x 驅(qū)動文件名
# 安裝
   sudo ./文件名 -no-x-check -no-nouveau-check -no-opengl-files

第十步開啟圖形化界面【若Xshell遠(yuǎn)程連接，不需要圖形化界面也可略過】

service lightdm start

第十一步查看是否安裝成功

nvidia-smi
# 出現(xiàn)如下返回則安裝成功

CUDA安裝

cuda官方下載地址

https://developer.nvidia.com/cuda-toolkit-archive

第一步下載相應(yīng)版本

第二步放到服務(wù)器上運(yùn)行安裝

sudo sh 安裝包名稱

輸入accept

去掉安裝驅(qū)動選項(xiàng)，并選擇Install【驅(qū)動上文已安裝】

第三步配置cuda環(huán)境變量(以下三種皆可)

export PATH="/usr/local/cuda-11.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

第四步更新環(huán)境變量

source ~/.bashrc

第五步查看是否安裝成功

nvcc -V

Cudnn安裝

cudnn下載官網(wǎng)地址

https://developer.nvidia.com/rdp/cudnn-download

第一步下載與cuda對應(yīng)版本

第二步解壓服務(wù)器上下載好的文件

tar -xvf 安裝包名字.tar.xz

第三步進(jìn)到解壓的文件夾內(nèi)部

# cp 后面指定自己下載的cuda版本

第四步至此已安裝完成

Anaconda安裝

Anaconda官方下載地址

https://www.anaconda.com/

第一步放到服務(wù)器上運(yùn)行安裝

sudo bash 安裝包名稱
# 空格跳過文字，選擇安裝路徑
# [/root/anaconda3] >>> /usr/anaconda3

第二步修改環(huán)境變量

第三步激活虛擬環(huán)境

多版本CUDA切換

第一步下載與cuda版本適配的gcc與g++

# 例如cuda10需要gcc4.8與g++4.8
sudo apt-get install gcc-4.8 g++-4.8
# 查詢已安裝版本
ls /usr/bin/gcc*

第二步進(jìn)行g(shù)cc與g++的版本切換

# gcc
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 40
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 90

# g++
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 40
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 90
# 數(shù)字40，90代表優(yōu)先級，數(shù)字越大優(yōu)先級越高
# 若要移除（否則不用執(zhí)行）
sudo update-alternatives --remove gcc /usr/bin/gcc-4.8
# 輸入切換命令（切換g++只需要將gcc替換成g++即可）
sudo update-alternatives --config gcc
# 輸入需要選擇的版本對應(yīng)序號（0,1,2）

第三步安裝cuda(步驟與上面一樣)

# 安裝完修改軟連接
# sudo 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
# 更新環(huán)境變量
# source ~/.bashrc

第四步安裝cudnn(步驟與上面一樣)

第五步切換cuda

# 刪除默認(rèn)鏈接
sudo rm -rf /usr/local/cuda
# 指定版本鏈接
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda

第六步查看版本是否切換成功

# 查看cuda版本
nvcc -V
# 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

on

刪除默認(rèn)鏈接

sudo rm -rf /usr/local/cuda

指定版本鏈接

sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda文章來源地址http://www.zghlxwxcb.cn/news/detail-602629.html

第六步查看版本是否切換成功

~~~python
# 查看cuda版本
nvcc -V
# 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

到了這里，關(guān)于NVIDIA顯卡驅(qū)動及CUDA相關(guān)安裝流程（包括多版本cuda切換）的文章就介紹完了。如果您還想了解更多內(nèi)容，請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！