码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2018-10-24 22:17:11      阅读:183      评论:0      收藏:0      [点我收藏+]

标签:export   get   oca   cal   不能   library   配置   x64   压缩   

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

标签:export   get   oca   cal   不能   library   配置   x64   压缩   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!