服务器上的CUDA版本为7.0,而跑程序至少需要7.5。因此重新安装CUDA
卸载旧版本的CUDA
1 | cd /usr/local/cuda-7.0/bin |
如果有提示卸载失败的文件或文件夹,可以自己手动删除掉。
卸载原来的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 | export CUDA_HOME=/usr/local/cuda-7.5 |
验证安装
1 | $ cd ~/NVIDIA_CUDA-7.5_Samples |
这里使用的是官方提供的sample,如果编译过程不报错,说明安装成功。可通过nvcc --version
指令查看CUDA的版本信息。
安装cudnn
通常只安装CUDA是不够的,还需要安装对应的cudnn,可以参考这篇教程。