关于Torch torchvision Python版本对应关系说明

更新时间：2022年05月11日 09:46:35 投稿：jingxian

这篇文章主要介绍了关于Torch torchvision Python版本对应关系说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

1. torch- torchvision- python版本对应关系

2. CUDA Toolkit 和PyTorch的对应关系

仅供参考

3. 安装说明

3.1 用anaconda安装pytorch

anaconda新建虚拟环境后，直接在pytorch官网官网链接找到“Install”按钮。这里一键搞定torch，torchvision，cudatoolkit等等。

前提需要安装好NVIDIA驱动。
不需要另外安装CUDA（笔者在没有单独安装CUDA情况下，成功运行了torch-gpu）。
和机器已经装好的CUDA不发生冲突。
anaconda会自动对应torch，torchvision，cudatoolkit等的版本。
这里一般下载最新的torch版本。

原因是anaconda将torch，torchvision，cudatoolkit等等都集成在虚拟环境里，统一管理依赖包。

如图所示

有镜像源的情况下，去掉末尾的“-c pytorch”会更快，否则经常容易下载超时。conda下载超时时长可以设置。

conda config --show
conda config --set remote_connect_timeout_secs 40
conda config --set remote_read_timeout_secs 100

conda config --show		#查看conda设置
conda config --set 		#设置对应的参数

3.2 不用Anaconda来管理环境

仔细对照torch、torchvision、cuda之间的对应关系，还有NVIDIA和CUDA版本的关系。

3.3 对NVIDIA驱动的要求

和NVIDIA驱动直接关联的是CUDA的版本。

如果安装的是CUDA=10.0.130,那么在Windows系统要求NVIDIA驱动大于411.31,不需要严格等于411.31。

如果小于411.31，则会报错显示驱动版本过旧。

笔者是将NVIDIA驱动更新至最大456.38，正常使用。

3.4 下载 .whl 文件离线安装

https://download.pytorch.org/whl/torch_stable.html

根据前面的对应关系，下载好适合你的版本的 torch 、torchvision。

cu102 # 表示CUDA=10.2
cp37 # 表示python=3.7
linux or win

下载好后，用pip安装，先cd 到下载的文件夹

pip install torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl
pip install torchvision-0.8.0-cp36-cp36m-win_amd64.whl

测试GPU版本的torch是否安装成功

(torch) D:\MyData\xiaCN\Desktop\Work\unbiased> python
Python 3.6.13 (default, Feb 19 2021, 05:17:09) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python点云地面点滤波(Progressive Morphological Filter)算法介绍（PCL库）
这篇文章主要介绍了python点云地面点滤波(Progressive Morphological Filter)算法介绍（PCL库）,了解膨胀/腐蚀这两个基础操作,可以通过对其进行简单组合来形成开/闭操作,需要的朋友可以参考下
2021-08-08
使用python itertools实现计算双十一满减凑单
一年一度的双十一又到了,在这样一个日子中,可能遇到一些问题,首先是“凑单”问题,本文将使用python中的itertools库解决这一问题,感兴趣的可以了解下
2024-11-11
Python实现二叉树结构与进行二叉树遍历的方法详解
二叉树是最基本的数据结构,这里我们在Python中使用类的形式来实现二叉树并且用内置的方法来遍历二叉树,下面就让我们一起来看一下Python实现二叉树结构与进行二叉树遍历的方法详解
2016-05-05
Python基础教程之利用期物处理并发
这篇文章主要给大家介绍了关于Python基础教程之利用期物处理并发的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧。
2018-03-03
PyTorch Autograd的核心原理和功能深入探究
本文深入探讨了PyTorch中Autograd的核心原理和功能,从基本概念、Tensor与Autograd的交互,到计算图的构建和管理,再到反向传播和梯度计算的细节,最后涵盖了Autograd的高级特性
2024-01-01
python实现中文输出的两种方法
这篇文章主要介绍了python实现中文输出的两种方法,实例分析了Python操作中文输出的技巧,需要的朋友可以参考下
2015-05-05
Python语言编写智力问答小游戏功能
这篇文章主要介绍了使用Python代码语言简单编写一个轻松益智的小游戏,代码简单易懂，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-10-10
Python练习-承压计算
这篇文章主要介绍了Python练习-承压计算，前面我们练习了Python购物单，这篇我们继续练习承压计算，和前篇文章一样还是问题描述开始，需要的小伙伴可以参考一下
2022-01-01
Python3 SSH远程连接服务器的方法示例
这篇文章主要介绍了Python3 SSH远程连接服务器的方法示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-12-12
Python FtpLib模块应用操作详解
这篇文章主要介绍了Python FtpLib模块应用操作,结合实例形式详细分析了FtpLib模块的功能、配置、应用相关操作技巧与使用注意事项,需要的朋友可以参考下
2019-12-12