Pytorch1.5.1版本安装的方法步骤


Posted in Python onDecember 31, 2020

查看自己cuda版本,

Pytorch1.5.1版本安装的方法步骤

我的cuda是11版本了,所以可以安装11版本以下的任何版本。

进入pytorch官网

官网网址:https://pytorch.org/

Pytorch1.5.1版本安装的方法步骤

2020年11月19号,更新

最简单的是直接按官网给的Run this Command命令,直接安装,如下:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

解释:-c pytorch,意思是从pytorch网站下载,速度感人,有办法的那就方便多了。

按照上面图这样选择,安装pytorch有GPU加速的版本,安装命令可以改下,后面加个豆瓣源,这样下载速度快些。

pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

或者直接用conda安装,去掉后面的 -c pytorch

conda install pytorch torchvision cudatoolkit=10.2

如果上面方法都下载慢,那就按下面方法来。(适用于win版本,Linux的可以返回上一层寻找对应的版本)

先进清华源https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=D
下载对应pytorch版本,我的是Python3.7,需要PyTorch-gpu版本,cuda需要10.2,找到对应的bz2文件下载,如图

Pytorch1.5.1版本安装的方法步骤

还要下载对应的torchvision===0.6.1,如图

Pytorch1.5.1版本安装的方法步骤

下载好就在命令行进入你下载的路径目录里面安装,并输入下面代码进行离线安装。

conda install --offline 对应的安装包文件名字

安装完后还要安装cudatoolkit=10.2

conda install cudatoolkit=10.2

然后运行测试代码:

# TEST
import torch
from torch.backends import cudnn

x = torch.Tensor([1.0])
xx = x.cuda()
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
print(xx)
print(cudnn.is_acceptable(xx))

结果:

1.5.1
10.2
True
tensor([1.], device='cuda:0')
True

安装成功!

GPU加速代码

import torch
import time

print(torch.__version__)
print(torch.cuda.is_available())

a = torch.randn(10000, 1000)
b = torch.randn(1000, 2000)

t0 = time.time()
c = torch.matmul(a, b) # 矩阵乘法
t1 = time.time()
print(a.device, t1 - t0, c.norm(2))

t0 = time.time()
c = torch.matmul(a, b) # 矩阵乘法
t1 = time.time()
print(a.device, t1 - t0, c.norm(2))

device = torch.device('cuda')
a = a.to(device)
b = b.to(device)

t0 = time.time()
c = torch.matmul(a, b) # 矩阵乘法
t2 = time.time()
print(a.device, t2 - t0, c.norm(2))

t0 = time.time()
c = torch.matmul(a, b)
t2 = time.time()
print(a.device, t2 - t0, c.norm(2))

结果:

1.5.1
True
cpu 0.13901472091674805 tensor(140929.9688)
cpu 0.16696977615356445 tensor(140929.9688)
cuda:0 0.22500324249267578 tensor(141330.6875, device='cuda:0')
cuda:0 0.003974437713623047 tensor(141330.6875, device='cuda:0')

运行两次是cuda有个预热的过程,第二次的时间明显减少了。和CPU相比,更快。

自动求导

代码:

import torch
from torch import autograd

x = torch.tensor(1.)
a = torch.tensor(1., requires_grad=True)
b = torch.tensor(2., requires_grad=True)
c = torch.tensor(3., requires_grad=True)

y = a ** 2 * x + b * x + c

print('before:', a.grad, b.grad, c.grad)
grads = autograd.grad(y, [a, b, c])
print('after :', grads[0], grads[1], grads[2])

结果:

before: None None None
after : tensor(2.) tensor(1.) tensor(1.)

可以看出pytorch比TensorFlow1.X好理解,适合人类思维,功能也都全。

到此这篇关于Pytorch1.5.1版本安装的方法步骤的文章就介绍到这了,更多相关Pytorch1.5.1版本安装内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python-基础-入门 简介
Aug 09 Python
python基于pygame实现响应游戏中事件的方法(附源码)
Nov 11 Python
python 文件操作删除某行的实例
Sep 04 Python
Python爬取个人微信朋友信息操作示例
Aug 03 Python
python实现从文件中读取数据并绘制成 x y 轴图形的方法
Oct 14 Python
Python根据文件名批量转移图片的方法
Oct 21 Python
python实现生成字符串大小写字母和数字的各种组合
Jan 01 Python
python同步windows和linux文件
Aug 29 Python
Python 图像对比度增强的几种方法(小结)
Sep 25 Python
浅谈Python3识别判断图片主要颜色并和颜色库进行对比的方法
Oct 25 Python
Python使用qrcode二维码库生成二维码方法详解
Feb 17 Python
OpenCV+Python3.5 简易手势识别的实现
Dec 21 Python
python爬虫基础之urllib的使用
Dec 31 #Python
python opencv实现直线检测并测出倾斜角度(附源码+注释)
Dec 31 #Python
Python实现Appium端口检测与释放的实现
Dec 31 #Python
python 基于selectors库实现文件上传与下载
Dec 31 #Python
python中Mako库实例用法
Dec 31 #Python
Python调用系统命令os.system()和os.popen()的实现
Dec 31 #Python
Python使用Opencv实现边缘检测以及轮廓检测的实现
Dec 31 #Python
You might like
通过PHP修改Linux或Unix口令的方法分享
2012/01/30 PHP
jQuery下扩展插件和拓展函数的写法(匿名函数使用的典型例子)
2010/10/20 Javascript
jquery中dom操作和事件的实例学习 仿yahoo邮箱登录框的提示效果
2011/11/30 Javascript
jquery ajax请求实例深入解析
2012/11/26 Javascript
js编写trim()函数及正则表达式的运用
2013/10/24 Javascript
Javascript四舍五入Math.round()与Math.pow()使用介绍
2013/12/27 Javascript
javascript在当前窗口关闭前检测窗口是否关闭
2014/09/29 Javascript
解决jQuery uploadify在非IE核心浏览器下无法上传
2015/08/05 Javascript
javaScript事件学习小结(四)event的公共成员(属性和方法)
2016/06/09 Javascript
jQuery操作cookie
2016/08/08 Javascript
使用Curl命令查看请求响应时间方法
2016/11/04 Javascript
javascript实现简单的ajax封装示例
2016/12/28 Javascript
原生js轮播特效
2017/05/18 Javascript
基于JavaScript实现百度搜索框效果
2020/06/28 Javascript
vue cli 3.0 搭建项目的图文教程
2019/05/17 Javascript
Python中不同进制互相转换(二进制、八进制、十进制和十六进制)
2015/04/05 Python
Python中字典的浅拷贝与深拷贝用法实例分析
2018/01/02 Python
Python中的几种矩阵乘法(小结)
2019/07/10 Python
python利用7z批量解压rar的实现
2019/08/07 Python
Python自动化导出zabbix数据并发邮件脚本
2019/08/16 Python
python bluetooth蓝牙信息获取蓝牙设备类型的方法
2019/11/29 Python
Python requests模块基础使用方法实例及高级应用(自动登陆,抓取网页源码)实例详解
2020/02/14 Python
python爬取音频下载的示例代码
2020/10/19 Python
PHP数据运算类型都有哪些
2013/11/05 面试题
英文版区域经理求职信
2013/10/23 职场文书
成考报名单位证明范本
2014/01/16 职场文书
火箭队口号
2014/06/18 职场文书
有限公司股东合作协议书
2014/10/29 职场文书
幼儿园小班见习报告
2014/10/31 职场文书
遗嘱继承权公证书
2015/01/26 职场文书
就业意向协议书
2015/01/29 职场文书
2015年度优秀员工获奖感言
2015/07/31 职场文书
推广普通话主题班会
2015/08/17 职场文书
看古人们是如何赞美老师的?
2019/07/08 职场文书
一文搞懂如何实现Go 超时控制
2021/03/30 Python
Python机器学习之KNN近邻算法
2021/05/14 Python