目标:
在linux下,普通用户安装cuda和cuDNN
Step1. 安装cuda
1)确认下自己需要安装的cuda版本
假如需要安装cuda10.0,那么在nvidia官网下载需要的cuda版本(网址:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal)
2)可以复制下载网址,然后在本地执行wget url 进行下载。
3)安装:在自己的目录下新建cuda10.0 目录;然后执行如下命令,并指定--toolkitpath=$HOME/cuda10.0:
sh cuda_10.0.130_410.48_linux.run --silent --toolkit --toolkitpath=$HOME/cuda10.0
参考:https://zhuanlan.zhihu.com/p/95939378
Step2. 下载cuDNN,并解压
需要下载与cuda版本对应的cuDNN,比如与cuda10对应的cuDNN版本为:cuDNN V7.6.5
下载地址:https://developer.nvidia.com/rdp/cudnn-download,(在下载前需要注册为Nvidia 开发用户)
解压下载下来的文件:cudnn-10.0-linux-x64-v7.6.5.32.tgz,解压后会在当前目录下得到cuda文件夹,其中的内容就是cuDNN的内容;
Step3.拷贝cuDNN头文件与相关库文件到第一步安装的cuda文件夹中($HOME/cuda10.0)
cp cuda/include/cudnn.h $HOME/cuda10.0/include/
cp cuda/lib64/libcudnn* $HOME/cuda10.0/lib64/
chmod a+r $HOME/cuda10.0/include/cudnn.h $HOME/cuda10.0/lib64/libcudnn*
Step4: 修改 ~/.bashrc 文件,配置cuda与cuDNN相关库文件的路径;
export CUDA_HOME="$HOME/cuda10.0"
export PATH="$PATH:$CUDA_HOME/bin"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$CUDA_HOME/lib64"
以上!