程序员经验分享-hwhale

Pytorch环境配置——cuda、、cudnn、torch、torchvision对应版本（最全）及安装方法

2023-11-19

一、查询可支持的最高cuda版本

在安装显卡驱动的前提下（显卡驱动安装方法），输入：

nvidia-smi

在这里插入图片描述
可以看到该电脑可以支持的cuda版本最高是11.4，驱动是向下兼容的，所以cuda版本小于等于11.4的都可以安装上。
（如果想安装的cuda版本不在可支持的cuda版本内，则可以考虑升级电脑的内核，从而实现对cuda高版本的安装。具体cuda版本及需要的内核详见：CUDA Toolkit and Corresponding Driver Versions ）

二、查看cuda、cudnn、pytorch、torchvision对应版本

表1 cuda与cudnn对应的版本
cuda cudnn
11.5 8.3.3、8.3.2、8.3.1、8.3.0、8.2.1、8.2.0
11.4 8.2.4、8.2.2、8.2.1、8.2.0
11.3 8.2.1、8.2.0
11.2 8.2.1、8.2.0、8.1.1、8.1.0
11.1 8.2.1、8.2.0、8.1.1、8.1.0、8.0.5、8.0.5、8.0.4
11.0 8.2.1、8.2.0、8.1.1、8.1.0、8.0.5、8.0.4、8.0.3、8.0.2、8.0.1
10.2 8.3.3、8.3.2、8.3.1、8.3.0、8.2.4、8.2.2、8.2.1、8.2.0、8.1.1、8.1.0、8.0.5、8.0.4、8.0.3、8.0.2、8.0.1、7.6.5
10.1 8.0.5、8.0.4、8.0.3、8.0.2、7.6.5、7.6.4、7.6.3、7.6.2、7.6.1、7.6.0、7.5.1、7.5.0
10.0 7.6.5、7.6.4、7.6.3、7.6.2、7.6.1、7.6.0、7.5.1、7.5.0、7.4.2、7.4.1、7.3.1、7.3.0
9.2 7.6.5、7.6.4、7.6.3、7.6.2、7.6.1、7.6.0、7.5.1、7.5.0、7.4.2、7.4.1、7.3.1、7.2.1、7.1.4、7.1.2
9.1 7.1.3、7.1.2、7.0.5
9.0 7.6.5、7.6.4、7.6.3、7.6.2、7.6.1、7.6.0、7.5.1、7.5.0、7.4.2、7.4.1、7.3.1、7.3.0、7.1.4、7.1.3、7.1.2、7.0.5、7.0.4
8.0 7.1.4、7.1.3、7.0.5、6.0、5.1、5
7.5 5.1、5
7.0 4、3
6.5 2、1

cudnn是用于深度神经网络的GPU加速库，cuda是平台，两者都需要安装才可以调用GPU加速。
cuda与cudnn的关系详解
最新 【NVIDIA官网】cuda与cudnn对应的版本

CUDAToolkit是NVIDIA的CUDA工具包,包含了CUDA的全部工具。CUDAToolkit与cuda版本相同。

表2 cuda、CUDAToolkit与pytorch对应的版本
cuda CUDAToolkit pytorch
11.3 11.3 1.10.1、1.10.0、1.9.1、1.9.0、1.8.1
11.1 11.1 1.10.0、1.9.1、1.9.0、1.8.1、1.8.0
11.0 11.0 1.7.1、1.7.0
10.2 10.2 1.10.1、1.10.0、1.9.1、1.9.0、1.8.1、1.8.0、1.7.1、1.7.0、1.6.0、1.5.1、1.5.0
10.1 10.1 1.7.1、1.7.0、1.6.0、1.5.1、1.5.0、1.4.0
10.0 10.0 1.2.0、1.1.0、1.0.1、1.0.0
9.2 9.2 1.7.1、1.7.0、1.6.0、1.5.1、1.5.0、1.4.0、1.2.0
9.0 9.0 1.1.0、1.0.1、1.0.0
8.0 8.0 1.0.0

最新 【pytorch官网】CUDAToolkit与pytorch对应的版本

表3 pytorch与torchvision对应的版本
pytorch torchvision
1.10.1 0.11.2
1.10.0 0.11.0
1.9.1 0.10.1
1.9.0 0.10.0
1.8.1 0.9.1
1.8.0 0.9.0
1.7.1 0.8.2
1.7.0 0.8.0
1.6.0 0.7.0
1.5.1 0.6.1
1.5.0 0.6.0
1.4.0 0.5.0
1.2.0 0.4.0
1.1.0 0.3.0
1.0.1 0.2.2
1.0.0 0.2.1

最新 【pytorch官网】pytorch与torchvision对应的版本

三、安装

3.1 Windows上安装

  1. Windows——安装cuda10.2和cudnn7.6.5（两者搭配稳定可靠）
  2. PyTorch——基于Windows环境下的安装教程

3.2 Ubuntu上安装

  1. Ubuntu 18.04——Anaconda + CUDA + Cudnn + Pytorch + TensorRT 深度学习环境配置

四、测试是否安装成功

终端输入：

python

import torch

torch.cuda.is_available()

如果输出为True，则代表安装成功。

五、小贴士

  1. cuda、cudnn需要去官网下载后按照教程安装，这样安装成功几率大，直接命令行安装失败风险很大。
  2. 安装指定版本的pytorch、torchvision、cudatoolkit，在输入安装指令的那一步，直接加上版本号即可。举例：conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)

Ubuntu

Pytorch

Windows

深度学习环境配置

Pytorch环境配置——cuda、、cudnn、torch、torchvision对应版本（最全）及安装方法 的相关文章

随机推荐

  • 经典C语言程序之 C/C++：计算两个整数的和

    经典C语言程序之 C C 计算两个整数的和 在C C 编程中 计算两个整数的和是一个经典的问题 本文将介绍如何使用C语言编写一个简单的程序来实现这一功能 首先 我们需要定义两个整数变量来存储用户输入的值 并用scanf函数从用户处获取这些值

  • VSCode如何设置终端工作目录

    文章目录 前言 固定工作目录 Terminal Here 注意 前言 相信大家在使用VSCode的时候 都会有如下难受的感觉 每次打开终端的时候工作目录都是用户目录 如果要执行命令还得cd到当前文件夹 十分麻烦 为了提高工作效率 有必要设置

  • 算法------大整数

    大整数 1 大整数加法 include

  • freebsd 做文件服务器,freebsd文件服务器

    freebsd文件服务器 内容精选 换一换 FTP SFTP连接适用于从线下文件服务器或ECS服务器上迁移文件到OBS或数据库 当前仅支持Linux操作系统的FTP 服务器 连接FTP或SFTP服务器时 他们的连接参数相同 如表1所示 FT

  • ＜Linux开发＞linux开发工具- 之-开发使用linux命令记录

    Linux开发 linux开发工具 之 开发使用linux命令记录 本文章主要记录开发过程中涉及使用的linux命令 1 查看磁盘大小分区情况 命令 df hl 可查看分区的路径 及空间大小使用情况 以及挂在点位置 2 查看指定目录文件的大

  • unity点击按钮，弹出image和Text做的弹出框

    让弹出框开始不显示 点击按钮后显示 在Hierarchy面板上新建Image 下面我命名为win 再在上面放置Image和Text组成弹出框 如下图 建立一个Message脚本 using System Collections using

  • [毕业设计]C++程序类内聚度的计算与存储

    目录 前言 课题背景和意义 实现技术思路 实现效果图样例 前言 大四是整个大学期间最忙碌的时光 一边要忙着备考或实习为毕业后面临的就业升学做准备 一边要为毕业设计耗费大量精力 近几年各个学校要求的毕设项目越来越难 有不少课题是研究生级别难度

  • PLSQL 基础语法

    01 PLSQL 语法 变量 if loop cursor exception 1 语法 在数据库服务器上保存的打断可执行方法 供其他开发人员调用 可以有效的减少数据库端和服务端的数据交互 提高效率 降低带宽消耗 语法格式 declare

  • 网络工程毕业设计题目100例

    文章目录 0 简介 1 如何选题 2 最新网络工程选题 2 1 Java web SSM 系统 2 2 大数据方向 2 3 人工智能方向 2 4 其他方向 4 最后 0 简介 丹成学长 搜集分享最新的网络工程专业毕设毕设选题 难度适中 适合

  • Microsemi Libero系列教程（七）——SmartDebug的使用

    SmartDebug简介 SmartDebug的使用 ActiveProbe使用

  • VMware无法创建共享文件夹

    1 Linux安装VM 1 chmod x VMware Workstation Full 14 1 3 9474260 x86 64 1 bundle 赋予执行权限 2 VMware Workstation Full 14 1 3 947

  • 二面字节被问到VARCHAR(M) 到底占用多少个字节？我跟面试官硬刚了半小时

    前言 这个问题其实很有迷惑性 问的是字节 不是字符 我们知道在计算机中只能存储二进制数据 所以要搞清楚这个问题 就要搞清楚下面2个问题 1 字节和字符的对应关系 2 varchar 到底能存多少个字节 为了搞清楚上面两个问题 又必须搞清楚m

  • jenkins中使用git遇到的坑

    error src refspec master does not match any root c32e20fd20e8 var jenkins home workspace DataPlatform git push u DataPla

  • RK3568 Android12 RK628编译报错问题

    Platform RK3568 OS Android 12 Kernel v4 19 206 SDK Version android 12 0 mid rkr1 问题 在dts中配置rk628时遇到编译报错 提示找不到rk628的label

  • Kubernets原理分解

    主节点 master 快速介绍 master也要装kubelet和kubeproxy 前端访问 UI CLI kube apiserver scheduler controller manager etcd kubelet kubeprox

  • 高考失利，还适合选计算机专业吗？？

    前言 高考落榜 人生陷入低谷 对于很多人来说 这意味着梦想的破灭和无尽的绝望 但是 对于我来说 这只是人生旅程的一个起点 我喜欢编程也热爱编程 虽然网上很多言论说计算机行业已经很卷了 但是我却认为无论再哪个行业 你不卷 也同样落后于人 所以

  • 微信小程序中上传图片添加水印功能

    功能实现 参考文章 https blog csdn net yynikki article details 101763718 遇到的问题 实现过程中主要遇到的问题有以下两个 1 在微信开发者工具上图片显示完全正常 但在真机上生成的图片不全

  • 理解一维卷积

    根据我个人的经验和偏好 理解数学概念的最好方式之一就是赋予其物理意义 把f t 看做输入 g t 看做系统的衰减系数 卷积就比较好理解了 在某一时刻n 该系统对f n 的响应值就是f n xg 0 但系统的总输出C n 不仅跟当前输入的f

  • HackBar 使用教程

    啥是Hackbar Hackbar是一个Firefox的插件 它的功能类似于地址栏 但是它里面的数据不受服务器的相应触发的重定向等其它变化的影响 有网址的载入于访问 联合查询 各种编码 数据加密功能 这个Hackbar可以帮助你在测试SQL

  • Pytorch环境配置——cuda、、cudnn、torch、torchvision对应版本（最全）及安装方法

    Pytorch环境配置 cuda cudnn torch torchvision对应版本 最全 及安装方法 一 查询可支持的最高cuda版本 二 查看cuda cudnn pytorch torchvision对应版本 三 安装 3 1 W

热门标签