Ubuntu上CUDA的安装
下载路径: CUDA下载
- `sudo dpkg -i cuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb`
- `sudo apt-get update`
- `sudo apt-get install cuda`
设置LD_LIBRARY_PATH:
最简单的方式在你的 /etc/profile 或 ~/.bashrc 文件中添加如下语句
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
或者执行命令 sudo vi /etc/ld.so.conf.d/cuda.conf,在文件中添加如下内容:
/usr/local/cuda/lib64
/lib
然后再执行命令:
sudo ldconfig -v
CUDNN
下载网址:CUDNN (需要注册申请,批准通过,才能下载)
缺省的编译不支持CUDNN,所以如果要添加CUDNN支持,要修改make/config.mk文件中的编译选项,使之支持CUDNN。
解压缩下载文件,将 include 和 lib64 拷贝放置在 /usr/local/cuda 目录下即可。我下载使用的是CUDNN V4版本
执行如下命令修改链接文件:
cd /usr/local/cuda/lib64 rm -rf libcudnn.so libcudnn.so.4 ln -s libcudnn.so.4.0.4 libcudnn.so.4 ln -s libcudnn.so.4 libcudnn.so