linux下普通用户安装CUDA和cuDNN

目标:

在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"

 

以上!

你可能感兴趣的:(计算机视觉,linux)