安装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
完成安装。