电脑配置:
ubuntu 14.04 64bit
128G 内存
GTX Titan X显卡
软件版本:
Cuda 7.0
Cudnn-7.0
1. 安装开发所需的依赖包
sudo apt-get install build-essential # basic requirement
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler #required by caffe
2.安装CUDA 7.0
安装CUDA有两种方法,离线.run安装,在线.deb安装。官网下载地址
切换到下载的deb所在目录,执行下边的命令
sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb
sudo apt-get update
sudo apt-get install cuda
我安装的是 :cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb
3.安装cudnn 7.0 v4
下载 cudnn-7.0-linux-x64-v4.0-prod.tgz,官网下载地址
注:要用学校邮箱注册,才被允许下载
下载后,切换到下载目录
tar -zxvf cudnn-7.0-linux-x64-v4.0-prod.tgz
cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
注:不同版本的cudnn解压出来的文件夹名称不一样,根据文件名称再执行后两句
更新软链接
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.4.0.7
sudo ln -sf libcudnn.so.4.0.7 libcudnn.so.4
sudo ln -sf libcudnn.so.4 libcudnn.so
sudo ldconfig
4,设置环境变量
在/etc/profile中添加CUDA环境变量
sudo gedit /etc/profile
加入以下两句