安裝 cuda 8.0
在ubuntu16.04 安裝caffe依賴(lài)于cuda 8, 因此需要安裝cuda 8 。
安裝顯卡驅(qū)動(dòng),直接在: system settings→software & updates→additional drivers 安裝就行可以了。
安裝 cuda toolkit 需要在命令行界面,首先需要關(guān)閉圖形界面:
sudo service lightdm stop在終端運(yùn)行指令 sudo sh cuda_8.0.27_linux.run
不需要安裝顯卡驅(qū)動(dòng),這個(gè)選no其他都是yes.安裝完成后,需要裝一些依賴(lài)庫(kù):
sudo apt-get install freeglut3-dev build-essential libx11-dev
libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa
libglu1-mesa-dev設(shè)置環(huán)境變量:
在終端輸入這兩句
export PATH=/usr/local/cuda-8.0/bin:$PATH
Export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
然后在 文件中 sudo vi /etc/profile 輸入上面這兩句話(huà),保存退出。
最后輸入:sudo ldconfig 使環(huán)境變量生效。驗(yàn)證是否完成:
nvidia-smi
nvcc-V
都會(huì)顯示出相關(guān)信息。
安裝 cudnn 5.0
安裝 cuDNN 比較簡(jiǎn)單,解壓后把相應(yīng)的文件拷貝到對(duì)應(yīng)的 CUDA 目錄下即可:
- 解壓:tar -zxvf cudnn-8.0-linux-x64-v5.0-ga.tgz
文件夾中有一下內(nèi)容:
*cuda/include/cudnn.h
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.5
cuda/lib64/libcudnn.so.5.0.5
cuda/lib64/libcudnn_static.a - 拷貝到相應(yīng)目錄,行以下指令:
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* - 驗(yàn)證一下
cuda 的 samples 里面有個(gè) deviceQuery 運(yùn)行之后會(huì)顯示信息,最后一行
出行 pass 說(shuō)明成功啦!