首页 > 其他 > 详细

机器学习环境配置系列二之cuDNN

时间:2018-10-24 21:01:58      阅读:237      评论:0      收藏:0      [点我收藏+]

1、下载cuDNN

  1. 前往: NVIDIA cuDNN home page.
  2. 进入下载
  3. 勾选Nvidia的协议复选框(流氓的选择,不勾选不能下载)
  4. 选择与安装的cuda版本一致的cudnn进行下载.

2、安装cuDNN

  • 解压文件包
$ tar -xzvf cudnn-9.0-linux-x64-v7.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*

  

3、路径配置(路径配置非常重要,这个决定了cudnn是否可以正确运行)

  • 在用户目录下执行命令
vi .bashrc
  • 添加内容如下
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
  • 运行命令
source ~/.bashrc

  

 

机器学习环境配置系列二之cuDNN

原文:https://www.cnblogs.com/jaww/p/9845923.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!