一. 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*