在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的版本等信息。
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,请发送邮件至 55@qq.com 举报,一经查实,本站将立刻删除。转转请注明出处:https://www.szhjjp.com/n/883842.html