关于Torch torchvision Python版本对应关系说明
作者:jingxian 发布时间:2021-06-17 09:13:52
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
来源:https://blog.csdn.net/qq_40630902/article/details/118356845


猜你喜欢
- 一,VSCode微软出厂的高颜值编辑器VSCode官网1.加载大文件几乎秒开,运行速度很快2.跨平台的文本编辑器,内置了对许多主流语言的支持
- 1、对继承的理解继承(Inheritance) :代码复用的高级抽象继承是面向对象设计的精髓之一实现了以类为单位的高级抽象级别代码复用继承是
- wxPython树控件介绍树(tree)是一种通过层次结构展示信息的控件,如下图所示是树控件示例,左窗口中是树控件,在wxPython中树控
- 利用pymongo包进行数据库的连接,使用xlrd包读取excel数据,由于二者数据结构的不同,要将excel格式数据转换为json格式数据
- 大家好,我是丁小杰!今天和大家分享Pandas中四种有关数据透视的通用函数,在数据处理中遇到这类需求时,能够很好地应对。pandas.mel
- 列表的逆序遍历a = [1,3,6,8,9]print("通过下标逆序遍历1:")for i in a[::-1]: &
- 因此,在我接触那么多种语言当中,asp是最不严格的一种,是对程序员要求最低的一种。 昨天测试了asp.net、php和asp的运行速度比较,
- strftime()函数的用法strftime()函数可以把YYYY-MM-DD HH:MM:SS格式的日期字符串转换成其它形式的字符串。s
- 前端开发环境多数基于Node.js,好处不多说了。但与使用Visual Studio开发的后端Asp.Net Core项目一起调试,却不是很
- 安装pyserialpip install pyserial查看可用的端口# coding:utf-8import serial.tools
- 前言Modbus协议是由Modicon公司(现在的施耐德电气Schneider Electric)推出,主要建立在物理串口、以太网TCP/I
- 本文为大家分享了mysql 8.0.15 安装配置图文教程,供大家参考,具体内容如下1.进官网,选择适合自己版本的linux下的MYSQL版
- 1. HADOOP背景介绍1.1 什么是HADOOP1.HADOOP是apache旗下的一套开源软件平台2.HADOOP提供的功
- 下边是我对部分内容的总结,里边偏向了T-SQL语句实现的总结,对于SQL Server Management Studio中对象管理器的操作
- 1、过年的时候在手机上下载了2048玩了几天,心血来潮决定用py写一个,刚开始的时候想用QT实现,发现依赖有点大。正好看到graphics.
- 1.首先安装 PyPDF2 库:pip install PyPDF22.然后保存下面文件(已带注释,具体实现请自己思考)import osi
- 1、打开Sqlserver,选择登录名下的账户右击点击属性2、右击点击属性查看强制过期是否被勾选上,如勾选上,会在一段时间后该账户不能正常使
- 1、通过将函数存储在独立的文件中,可隐藏程序代码的细节,将重点放在程序的高层逻辑上。2、能让你在众多不同的程序中重用函数。将函数存储在独立文
- 调用数据库存储过程见下:<%Set Dataconn = Server.CreateObject(&qu
- 相关文章Pytest框架之fixture详解(一)Pytest框架之fixture详解(二)Pytest框架之fixture详解(三)本文关