首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >深度学习之CUDA + cudnn

深度学习之CUDA + cudnn

作者头像
数据科学工厂
发布于 2023-02-27 04:32:59
发布于 2023-02-27 04:32:59
43500
代码可运行
举报
运行总次数:0
代码可运行

导读

本文将介绍如何在Ubuntu下安装深度学习需要使用的CUDAcudnn

1. CUDA安装

  1. 前往官网[1]下载符合自己系统版本的CUDA,根据自己电脑配置和系统情况,选择对应的选项,如下图:

CUDA

  1. 根据网页的安装提示操作,如下图:

安装提示

  1. 打开终端进入到CUDA安装文件所在文件夹,赋予其可执行权利
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
sudo  chmod +x cuda_10.0.130_410.48_linux.run
  1. 安装
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
sudo ./cuda_10.0.130_410.48_linux.run
  1. 修改环境变量
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
sudo vi ~/.bashrc

export CUDA_HOME=/usr/local/cuda-10.0 # 根据版本修改
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

# 更新生效
source ~/.bashrc
  1. 查看安装的版本信息
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
nvcc -V
# cat /usr/local/cuda/version.txt

nvcc信息

2. cudnn安装

  1. 官网[2]下载与CUDA对应的cudnn版本(要注册账号才能下载)
  2. 打开终端进入cudnn下载目录,对其进行解压
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8
mv cudnn-10.0-linux-x64-v7.4.2.24.solitairetheme8 cudnn
  1. 将解压后的文件复制到cuda目录下
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
sudo cp cudnn/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
  1. 查看版本 (可能不显示,但不影响使用)
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

参考资料

[1]

CUDA官网: https://developer.nvidia.com/cuda-downloads

[2]

cudnn: https://developer.nvidia.com/cudnn

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2022-11-01,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 冷冻工厂 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 导读
  • 1. CUDA安装
  • 2. cudnn安装
    • 参考资料
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档