更改系统中的CUDA版本

服务器上的CUDA版本为7.0,而跑程序至少需要7.5。因此重新安装CUDA

卸载旧版本的CUDA

1
2
cd /usr/local/cuda-7.0/bin
sudo ./uninstall_cuda_toolkit_7.0.pl

如果有提示卸载失败的文件或文件夹,可以自己手动删除掉。

卸载原来的NVIDIA驱动

sudo apt-get remove nvidia*

安装CUDA

安装开发需要的基本包
sudo apt-get install build-essential
前往Nvidia官网下载CUDA安装包。

1
2
3
$ sudo dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

安装完成,可以重启机器以验证安装。

添加环境变量

1
2
3
4
export CUDA_HOME=/usr/local/cuda-7.5
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
PATH=${CUDA_HOME}/bin:${PATH}
export PATH

验证安装

cuda-install-samples-7.5.sh ~
1
2
3
$ cd ~/NVIDIA_CUDA-7.5_Samples
$ cd 1_Utilities/deviceQuery
$ make

这里使用的是官方提供的sample,如果编译过程不报错,说明安装成功。可通过nvcc --version指令查看CUDA的版本信息。

安装cudnn

通常只安装CUDA是不够的,还需要安装对应的cudnn,可以参考这篇教程