关于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
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇关于sys.stdout和print的区别详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要为大家详细介绍了python读取LMDB中图像的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-07-07
这篇文章主要介绍了python如何实现代码检查,如果代码不规范，乱七八糟的，被小伙伴打死或者熬夜找bug事小，影响项目进度或者破坏项目结构就完蛋了,需要的朋友可以参考下2019-06-06
这篇文章主要介绍了TensorFlow2实现线性回归的详细解析，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-09-09
这篇文章主要介绍了Python接口自动化判断元素原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-02-02
这篇文章主要介绍了pandas如何获取某个数据的行号问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-02-02
Python Pillow Image.save 保存为jpg图片压缩问题
Pillow 库支持多种图片格式，Pillow 能够很轻松地实现图片格式之间的转换。本文就来详细的介绍一下Image.save的具体使用，感兴趣的可以了解一下2021-11-11
这篇文章主要介绍了python3 深浅copy对比详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
网络设备配置备份与恢复在网络安全管理中起着至关重要的作用,本文为大家介绍了如何通过Python实现网络设备配置备份与恢复,需要的可以参考下2025-03-03
本文以Python初级程序员为对象,介绍了如何使用asyncio和logging模块实现一个异步批处理文件的并发处理系统,以提高处理大量文件或数据时的效率,其中,通过配置日志系统记录处理文件的日志信息,定义AsyncBatchProcessor类控制并发任务的数量2024-10-10
最新评论