ubuntu中cudnn安装的方法是什么

avatar
作者
猴君
阅读量:1

在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的版本等信息。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!