Ubuntu 16.04下 cuda 8.0+cuDNN 5.1 安裝

cudaNVIDIA的編程語言平臺,想使用GPU就必須要使用cuda

cuDNNGPU加速計算深層神經網絡的庫。



環境: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"

測試CUDASamples

執行以下命令

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

如果顯示的是一些關於GPU的信息,則說明安裝成功了。

 

下載cuDNN

 

在官網下載cuDNN5.1https://developer.nvidia.com/rdp/cudnn-archive(注意CUDAcuDNN相對應的版本)


安裝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

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