1.首先查看自己的电脑最高支持cuda版本号是多少
nvidia-smi
2.然后到英伟达官网下载cuda
这是我第一次下载的时候选择的runfile(local),一定要根据自己电脑情况来选择,将下面的指令复制到终端下载即可。
3.遇到的问题
段错误(核心已转储)
第一种解决方法:考虑是栈溢出
终端输入ulimit -a查看用户界限,会发现stack size为8192(也可能是其他值)
输入ulimit -s unlimited将栈修改为无限
第二种解决方法:
若修改之后还是没有结局,可以使用wget -c从错误处继续下载
第三种解决方法:
若还是没有结局,那么可以把在官网下载cuda时候的Installer Type从runfile改成deb(local)
输入指令下载即可,输入指令的时候一定要小心,否则会出先玄幻的报错。
下载结束之后更改环境变量,把CUDA相关命令和库文件添加到系统目录中,不然会出现有些命令找不到的情况,全部更改结束之后终端输入nvcc -V出现下图则表示安装成功
3.安装cudnn
cuDNN Archive | NVIDIA Developer(下载链接)
进入官网下载,一定要下载对应版本的
找到和cuda版本对应的cudnn版本
建议下载这三个 ,得到三个安装包,直接解压安装
进入安装目录然后挑一个编译,比如conv_sample
cd conv_sample
sudo make
./conv_sample
最后出现下图
TEST PASSED
安装成功!
由于本人也是经验不足,如有错误欢迎各位斧正!