前提條件: nvidia驅動已經安裝好。詳見:Ubuntu如何查看顯卡信息及安裝nvidia顯卡驅動
一 CUDA的安裝
- 查看安裝的nvidia驅動版本號
nvidia-smi //第一行信息
- 選擇安裝的CUDA版本並下載
下載地址
英偉達官方的cuda和驅動的對應
- 安裝CUDA
sudo sh cuda_8.0.61_375.26_linux.run
就進入安裝了,開始的一大堆文字都是End User License Agreement,可以CTRL +C 跳過,然後輸入accept表示接受協議。
Do you accept the previously read EULA?
accept/decline/quit:accept
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?
(y)es/(n)o/(q)uit: n
Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y
Enter Toolkit Location
[ default is /usr/local/cuda-8.0 ]:
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y
Enter CUDA Samples Location
[ default is /home/kinny ]:
Installing the CUDA Toolkit in /usr/local/cuda-8.0 ...
Missing recommended library: libXmu.so
Installing the CUDA Samples in /home/kinny ...
Copying samples to /home/kinny/NVIDIA_CUDA-8.0_Samples now...
Finished copying samples.
===========
= Summary =
===========
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-8.0
Samples: Installed in /home/kinny, but missing recommended libraries
Please make sure that(環境變量按如下方式添加)
- PATH includes /usr/local/cuda-8.0/bin
- LD_LIBRARY_PATH includes /usr/local/cuda-8.0/lib64, or, add /usr/local/cuda-8.0/lib64 to /etc/ld.so.conf and run ldconfig as root
To uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-8.0/bin
Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-8.0/doc/pdf for detailed information on setting up CUDA.
***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 361.00 is required for CUDA 8.0 functionality to work.
To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:
sudo <CudaInstaller>.run -silent -driver
Logfile is /tmp/cuda_install_17494.log
注意:
安裝時可能出現錯誤:cuda:Missing recommended library: libGLU.so,libX11.so,libXi.so,libXmu.so
解決方案:方案1、方案2、方案3
(這個錯誤我沒解決,不知道後續有沒有影響)
4. 配置Cuda環境變量:
sudo su //切換爲root超級用戶的命令
sudo gedit ~/.bashrc //打開.bashrc文件
在 ~/.bashrc 的最後添加:
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
配置完環境變量之後,一定要更新一下,否則不能立即生效。也可以通過重啓電腦使得環境變量生效:
source ~/.bashrc
nvcc --version //檢查安裝是否成功
如果顯示下面的文字就說明安裝成功了:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Tue_Jun_12_23:07:04_CDT_2018
Cuda compilation tools, release 9.2, V9.2.148
su username //可切換普通用戶的命令
二 CUDA的安裝
- 下載cuDNN
首先到官網下載cuDNN( https://developer.nvidia.com/cudnn ),需要註冊爲開發者才能下載,而且直接下載可能速度非常慢,如果有別人已經下載過的,應該是可以直接拿過來用,不過前提是版本是對應的。 - 文件名爲:cudnn-××××.tgz,解壓:
tar xvzf cudnn-9.2-linux-x64-v7.6.4.38.tgz
- 解壓後可以看到cuda文件夾,在當前目錄打開終端,然後將庫和頭文件copy到cuda目錄並修改文件訪問權限:
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*
- 檢查查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果顯示下面的文字就說明安裝成功了:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 4
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"