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计算最大优先级队列实例
Dec 18 Python
python中正则表达式的使用详解
Oct 17 Python
Python中property函数用法实例分析
Jun 04 Python
对python字典过滤条件的实例详解
Jan 22 Python
详解python中的数据类型和控制流
Aug 08 Python
python word转pdf代码实例
Aug 16 Python
python错误调试及单元文档测试过程解析
Dec 19 Python
关于python pycharm中输出的内容不全的解决办法
Jan 10 Python
python圣诞树编写实例详解
Feb 13 Python
Python操作Jira库常用方法解析
Apr 10 Python
Django修改app名称和数据表迁移方案实现
Sep 17 Python
用Python实现一个打字速度测试工具来测试你的手速
May 28 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
PHP4实际应用经验篇(8)
2006/10/09 PHP
Android ProgressBar进度条和ProgressDialog进度框的展示DEMO
2013/06/19 PHP
获取URL文件名后缀
2013/10/24 PHP
使用ThinkPHP+Uploadify实现图片上传功能
2014/06/26 PHP
phpnow php探针环境检测代码
2014/11/04 PHP
PHP5.3新特性小结
2016/02/14 PHP
php获取远程图片并下载保存到本地的方法分析
2016/10/08 PHP
Laravel中encrypt和decrypt的实现方法
2017/09/24 PHP
PHP给源代码加密的几种方法汇总(推荐)
2018/02/06 PHP
让你的博文自动带上缩址的实现代码,方便发到微博客上
2010/12/28 Javascript
div模拟滚动条效果示例代码
2013/10/16 Javascript
javascript与Python快速排序实例对比
2015/08/10 Javascript
jquery实现ajax加载超时提示的方法
2016/07/23 Javascript
JavaScript中的子窗口与父窗口的互相调用问题
2017/02/08 Javascript
vue项目中的webpack-dev-sever配置方法
2017/12/14 Javascript
vue实现底部菜单功能
2018/07/24 Javascript
富文本编辑器vue2-editor实现全屏功能
2019/05/26 Javascript
vue 使用element-ui中的Notification自定义按钮并实现关闭功能及如何处理多个通知
2019/08/17 Javascript
使用Layui搭建后台管理界面的操作方法
2019/09/20 Javascript
[01:14]DOTA2 7.22版本新增神杖效果展示(智力英雄篇)
2019/05/29 DOTA
Python2.5/2.6实用教程 入门基础篇
2009/11/29 Python
python内存管理分析
2015/04/08 Python
深入浅析python定时杀进程
2016/06/06 Python
用 Python 爬了爬自己的微信朋友(实例讲解)
2017/08/25 Python
python使用tensorflow深度学习识别验证码
2018/04/03 Python
PyTorch CNN实战之MNIST手写数字识别示例
2018/05/29 Python
python实现决策树分类
2018/08/30 Python
PyCharm代码提示忽略大小写设置方法
2018/10/28 Python
python中web框架的自定义创建
2019/09/08 Python
CSS3实现时间轴特效
2020/11/02 HTML / CSS
中考冲刺决心书
2014/03/11 职场文书
幼儿园保育员岗位职责
2014/04/13 职场文书
小学六年级学生评语
2014/04/22 职场文书
民族精神月活动总结
2014/08/28 职场文书
公司的门卫岗位职责
2014/09/09 职场文书
老公给老婆的检讨书(精华篇)
2014/10/18 职场文书