ubuntu 16.04安裝nvdia驅動,cuda驅動以及cudnn

一. nvidia驅動安裝

1. 查詢顯卡型號

nvidia-smi -L

出現顯卡型號,我的是:GPU 0: GeForce RTX 2080 SUPER

2. 查詢顯卡驅動

網址:https://www.geforce.cn/drivers
輸入顯卡型號並搜索,列表中會出現很多驅動版本,選擇一款驅動

3.安裝顯卡驅動

網上常見的做法是:從命令行系統安裝,巨麻煩,推薦使用apt-get install

sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update

sudo apt-get install nvidia- table鍵補全,安裝你要的驅動即可

sudo apt-get install nvidia-430

4.查看顯卡驅動

nvidia-smi

二. cuda安裝

cuda對於nvidia驅動是有要求的,具體版本需要請見:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
在這裏插入圖片描述
cuda驅動下載地址:https://developer.nvidia.com/cuda-toolkit-archive

注意cuda對gcc版本也是有要求的:

從CUDA 7.5版開始,完全支持gcc 4.8,在Ubuntu 14.04和Fedora 21上支持4.9。
從CUDA 8版本開始,Ubuntu 16.06和Fedora 23完全支持gcc 5.3。
從CUDA 9版本開始,Ubuntu 16.04,Ubuntu 17.04和Fedora 25完全支持gcc 6。

具體安裝請參見, 圖形界面安裝即可:https://blog.csdn.net/briliantly/article/details/79560477

查看驅動是否安裝成功:

nvcc -V

我這裏輸出的信息爲:

nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2017 NVIDIA
Corporation Built on Fri_Nov__3_21:07:56_CDT_2017 Cuda compilation
tools, release 9.1, V9.1.85

三. cuDNN安裝

cuDNN對cuda版本也有需求,具體參見:https://developer.nvidia.com/rdp/cudnn-archive
安裝非常簡單,複製粘貼即可:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include sudo cp
cuda/lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r
/usr/local/cuda/include/cudnn.h sudo chmod a+r
/usr/local/cuda/lib64/libcudnn*

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章