Linux 拨号vps windows公众号手机端

ubuntu中cudnn安装的方法是什么

lewis 9年前 (2016-12-30) 阅读数 9 #程序编程
文章标签 ubuntu

在Ubuntu中安装cuDNN(CUDA Deep Neural Network library)可以按照以下步骤进行:

  1. 首先,需要去NVIDIA官网注册并下载cuDNN的压缩包,地址为:https://developer.nvidia.com/cudnn

  2. 下载完成后,将压缩包解压缩到指定目录,例如解压到/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是根据你下载的具体版本号和系统架构来确定的,需要根据实际情况进行更改。

  1. 配置环境变量,在.bashrc.zshrc等文件中添加如下路径:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
  1. 最后,运行以下命令使变量生效:
source ~/.bashrc

完成以上步骤后,cuDNN就成功安装到系统中了。可以检查是否安装成功,以及cuDNN的版本等信息。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门