转https://blog.csdn.net/hizengbiao/article/details/88625044
cuda 10.1:
按空格根据需要选择要安装的模块,有“X”的表示安装,没有的表示不安装,我这里已经安装过418.116的显卡驱动了,所以选择不安装驱动:
按上下键移动到CUDA Toolkit 10.1上,然后按“A”键,出现:
全部不安装并更改Toolkit Install Path:
再返回主界面然后选择Options->Library install path:
然后确定,返回选择Install
注:若出现 Error: unsupported compiler: X.X.X,一般是gcc版本和CUDA没对应,可尝试指令sh cuda_8.0.61_375.26_linux-run --override,如果没解决,则需要安装相对应的gcc。
配置cuDNN
使用指令tar -xzvf /home/zb/cuda/cudnn-8.0-linux-x64-v7.1.tgz -C /home/zb/cuda/tem/ 将cuDNN解压到tem文件夹并执行以下指令3:
cp /home/zb/cuda/tem/cuda/include/cudnn.h /home/zb/cuda/cuda-8.0/include/
cp /home/zb/cuda/tem/cuda/lib64/libcudnn* /home/zb/cuda/cuda-8.0/lib64
chmod a+r /home/zb/cuda/cuda-8.0/include/cudnn.h /home/zb/cuda/cuda-8.0/lib64