在没有root访问权限的情况下为Theano安装cuDNN

时间:2016-09-01 03:55:49

标签: python linux gpu theano-cuda

我可以在没有超级用户权限的情况下在本地安装cuDNN吗?

我没有root权限访问我正在使用的linux机器(发行版是openSuse),但我已经安装了CUDA 7.5。

我正在使用Theano,我需要cuDNN来提高GPU上的操作速度。

我从Nvidia下载了cudnn-7.5-linux-x64-v5.1,根据说明,我需要将CuDNN存档内容复制到CUDA安装文件夹,即(cuda / lib64 /和cuda / include /)。但这需要我有root权限。

我是否有可能在本地提取cudnn存档并为theano提供cudnn库的路径?

1 个答案:

答案 0 :(得分:6)

您可以将整个CUDA SDK复制到您的家中,并通过在~/.bashrc

中添加/修改这些环境变量,告诉Theano和其他人他们应该使用您的本地CUDA副本
export CUDA_ROOT=~/program/cuda-7.5
export CUDA_HOME=~/program/cuda-7.5
export PATH=${CUDA_HOME}/bin:$PATH
export LD_LIBRARY_PATH=/usr/lib64/nvidia:${CUDA_HOME}/lib64:$LD_LIBRARY_PATH

然后,您只需将cuDNN提取到本地CUDA SDK目录~/program/cuda-7.5/

即可