阅读量:1
在Ubuntu中安装cuDNN(CUDA Deep Neural Network library)可以按照以下步骤进行:
首先,需要去NVIDIA官网注册并下载cuDNN的压缩包,地址为:https://developer.nvidia.com/cudnn
下载完成后,将压缩包解压缩到指定目录,例如解压到
/usr/local
目录下:
tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz sudo cp cuda/include/cudnn.h /usr/local/cuda/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
注意:这里的cudnn-10.0-linux-x64-v7.6.5.32.tgz
是根据你下载的具体版本号和系统架构来确定的,需要根据实际情况进行更改。
- 配置环境变量,在
.bashrc
或.zshrc
等文件中添加如下路径:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
- 最后,运行以下命令使变量生效:
source ~/.bashrc
完成以上步骤后,cuDNN就成功安装到系统中了。可以检查是否安装成功,以及cuDNN的版本等信息。