安裝CUDA
1-首先對系統進行更新
sudo apt-get update
sudo apt-get upgrade
2-安裝依賴項
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
3-下載CUDA8.0,runfile類型的文件,更改權限並安裝。之前已安裝了驅動,此時選擇不安裝驅動,其他選擇默認。
sudo chmod a+x cuda_8.0.61_375.26_linux.run
sudo sh cuda_8.0.61_375.26_linux.run
有一個說明文件,需要一直按回車鍵,直到進度條顯示100%
4-配置環境變量
sudo gedit ~/.bashrc
在末尾添加如下內容,保存後退出。
export CUDA_HOME=/usr/local/cuda-8.0
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-8.0/bin:$PATH
刷新
source ~/.bashrc
5-創建鏈接文件
sudo gedit /etc/ld.so.conf.d/cuda.conf
添加如下內容
/usr/local/cuda-8.0/lib64
保存後關閉,使立即執行
sudo ldconfig
6-測試結果
使用nvcc --version,測試是否安裝成功。
安裝cuDNN
1-下載cuDNN,並解壓縮
tar xzvf cudnn-8.0-linux-x64-v6.0.0
2-拷貝解壓後的文件,並更改權限
sudo cp -a cuda/include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*
3-切換目錄操作
cd /usr/local/cuda-8.0/lib64/
刪除原動態文件(如果有) sudo rm -rf libcudnn.so libcudnn.so.5
生成軟鏈接 sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so
以下原理同上
sudo rm -rf libcudnn.so libcudnn.so.6
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so
完成安裝。