要在Ubuntu上配置cuDNN,首先需要确保已经安装了NVIDIA的GPU驱动和CUDA。然后按照以下步骤进行配置:
下载cuDNN:首先需要登录NVIDIA的开发者网站(https://developer.nvidia.com/cudnn),注册并下载适用于你的CUDA版本的cuDNN压缩包。
解压缩cuDNN:解压下载的压缩包,并将其中的文件复制到CUDA的安装目录下。一般情况下,CUDA的安装路径为/usr/local/cuda/
。
设置环境变量:编辑~/.bashrc
文件,并添加以下几行代码:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
保存并退出文件,然后运行以下命令使配置生效:
source ~/.bashrc
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果成功显示cuDNN的版本信息,则表示配置成功。
现在,您已经成功在Ubuntu上配置了cuDNN,可以开始在深度学习项目中使用它了。