cuda是NVIDIA的編程語言平臺,想使用GPU就必須要使用cuda。
cuDNN是GPU加速計算深層神經網絡的庫。
環境:Ubuntu 16.04 64bit
Nvidia GeForce GTX 960M (驅動Nvidia-Linux-x86_64-384.130)
安裝:cuda 8.0.44
cuDNN 5.1
顯卡驅動安裝:https://blog.csdn.net/feishicheng/article/details/70662094
鏈接:https://pan.baidu.com/s/1dAv_HKIWy6Kk5kwrIIza7A密碼:olku
下載CUDA
去官網https://developer.nvidia.com/cuda-downloads下載CUDA。
我選擇的是:
Operating System : Linux
Architecture : x86_64
Distribution : Ubuntu
Version : 16.04
Installer Type : runfile(local)
安裝CUDA
安裝:
下載後即可開始安裝CUDA,這裏說明一點,在安裝CUDA時也可以安裝NVIDIA驅動,但是一般選擇先安裝NVIDIA驅動,我安裝的是NVIDIA 384版本。
執行以下命令:
sudo bash cuda_8.0.44_linux.run
安裝過程中只需要注意在出現
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 361.62
選擇否即可,其餘的都直接默認或者選擇是。
配置環境變量:
執行命令
sudo vim ~/.bashrc
如果提示沒有安裝vim,則執行以下命令安裝vim
sudo apt-get install vim
安裝完後,可以使用以下命令查看
update-alternatives--display vi #已經添加到系統
vim --version #查看版本
進入.bashrc後,在末尾處添加
export PATH="/usr/local/cuda-8.0/bin:$PATH"
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:&LD_LIBRARY_PATH"
測試CUDA的Samples :
執行以下命令
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
如果顯示的是一些關於GPU的信息,則說明安裝成功了。
下載cuDNN
在官網下載cuDNN5.1。https://developer.nvidia.com/rdp/cudnn-archive(注意CUDA與cuDNN相對應的版本)
安裝cuDNN
下載完成後,進入相應的目錄,解壓文件。
確保cuda的安裝目錄是/usr/local/cuda/,之前安裝cuda時默認安裝目錄是這個,然後執行以下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cpcuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
執行完上述命令後,如果是系統中沒有其他版本的cuDNN,則安裝完畢,可以執行以下命令查看cuDNN的版本:
1sudo cat /usr/local/cuda/include/cudnn.h |grep CUDNN_MAJOR-A2