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中的函数编程
Apr 13 Python
详解JavaScript编程中的window与window.screen对象
Oct 26 Python
django session完成状态保持的方法
Nov 27 Python
opencv与numpy的图像基本操作
Mar 08 Python
python3.4 将16进制转成字符串的实例
Jun 12 Python
Django中的静态文件管理过程解析
Aug 01 Python
利用OpenCV和Python实现查找图片差异
Dec 19 Python
python函数调用,循环,列表复制实例
May 03 Python
Python + opencv对拍照得到的图片进行背景去除的实现方法
Nov 18 Python
python 实现汉诺塔游戏
Nov 28 Python
基于PyTorch实现一个简单的CNN图像分类器
May 29 Python
Python基本数据类型之字符串str
Jul 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图片上传代码
2013/11/04 PHP
js和php邮箱地址验证的实现方法
2014/01/09 PHP
php rsa加密解密使用详解
2015/01/14 PHP
Thinkphp整合微信支付功能
2016/12/14 PHP
PDO::exec讲解
2019/01/28 PHP
一个JavaScript继承的实现
2006/10/24 Javascript
jquery左右滚动焦点图banner图片鼠标经过显示上下页按钮
2013/10/11 Javascript
Jquery 在页面加载后执行的几种方式
2014/03/14 Javascript
jquery 判断滚动条到达了底部和顶端的方法
2014/04/02 Javascript
jquery+easeing实现仿flash的载入动画
2015/03/10 Javascript
介绍一个简单的JavaScript类框架
2015/06/24 Javascript
如何用jQuery实现ASP.NET GridView折叠伸展效果
2015/09/26 Javascript
Nodejs实战心得之eventproxy模块控制并发
2015/10/27 NodeJs
Angular设置title信息解决SEO方面存在问题
2016/08/19 Javascript
jQuery Validate验证框架详解(推荐)
2016/12/17 Javascript
详解VUE中v-bind的基本用法
2017/07/13 Javascript
微信小程序CSS3动画下拉菜单效果
2018/11/04 Javascript
Angular6 Filter实现页面搜索的示例代码
2018/12/02 Javascript
Vue数组响应式操作及高阶函数使用代码详解
2020/08/01 Javascript
在Vue中使用Select选择器拼接label的操作
2020/10/22 Javascript
[02:40]DOTA2殁境神蚀者 英雄基础教程
2013/11/26 DOTA
Python基于列表模拟堆栈和队列功能示例
2018/01/05 Python
python 对key为时间的dict排序方法
2018/10/17 Python
numpy 计算两个数组重复程度的方法
2018/11/07 Python
手把手教你将Flask应用封装成Docker服务的实现
2020/08/19 Python
Python安装Bs4的多种方法
2020/11/28 Python
Python Spyder 调出缩进对齐线的操作
2021/02/26 Python
eVitamins日本:在线购买折扣维生素、补品和草药
2019/04/04 全球购物
大专生简历的自我评价
2013/11/26 职场文书
门卫岗位安全职责
2013/12/13 职场文书
演讲稿怎么写才完美
2014/01/02 职场文书
村干部培训班主持词
2014/03/28 职场文书
国旗下的讲话演讲稿
2014/05/08 职场文书
汶川大地震感悟
2015/08/10 职场文书
CSS 还能这样玩?奇思妙想渐变的艺术
2021/04/27 HTML / CSS
升级 Win11 还是坚守 Win10?微软 Win11 新系统缺失功能大盘点
2022/04/05 数码科技