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单线程实现多个定时器示例
Mar 30 Python
Python实现遍历windows所有窗口并输出窗口标题的方法
Mar 13 Python
python判断给定的字符串是否是有效日期的方法
May 13 Python
解析Mac OS下部署Pyhton的Django框架项目的过程
May 03 Python
python使用pymysql实现操作mysql
Sep 13 Python
Python数据可视化编程通过Matplotlib创建散点图代码示例
Dec 09 Python
python获取代码运行时间的实例代码
Jun 11 Python
对Python中内置异常层次结构详解
Oct 18 Python
如何用python写一个简单的词法分析器
Dec 18 Python
python中的&&及||的实现示例
Aug 07 Python
python爬虫中采集中遇到的问题整理
Nov 27 Python
想学画画?python满足你!
Dec 24 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定时执行任务实现方法详解(Timer)
2015/07/30 PHP
基于PHP微信红包的算法探讨
2016/07/21 PHP
laravel-admin 中列表筛选方法
2019/10/03 PHP
javascript获取url上某个参数的方法
2013/11/08 Javascript
JS比较两个时间大小的简单示例代码
2013/12/20 Javascript
利用进制转换压缩数字函数分享
2014/01/02 Javascript
jquery实现ajax提交form表单的方法总结
2014/03/03 Javascript
javascript数组操作(创建、元素删除、数组的拷贝)
2014/04/07 Javascript
EasyUi中的Combogrid 实现分页和动态搜索远程数据
2016/04/01 Javascript
Node.JS 循环递归复制文件夹目录及其子文件夹下的所有文件
2017/09/18 Javascript
基于javascript中的typeof和类型判断(详解)
2017/10/27 Javascript
cropper js基于vue的图片裁剪上传功能的实现代码
2018/03/01 Javascript
Vue-Router的使用方法
2018/09/05 Javascript
微信小程序swiper实现滑动放大缩小效果
2018/11/15 Javascript
基于jquery实现九宫格拼图小游戏
2018/11/30 jQuery
微信小程序wepy框架学习和使用心得详解
2019/05/24 Javascript
详解微信小程序开发(项目从零开始)
2019/06/06 Javascript
详解Node.js异步处理的各种写法
2019/06/09 Javascript
JavaScript实现简单贪吃蛇效果
2020/03/09 Javascript
[51:50]完美世界DOTA2联赛 Magma vs GXR 第一场 11.07
2020/11/10 DOTA
python使用正则表达式分析网页中的图片并进行替换的方法
2015/03/26 Python
用matplotlib画等高线图详解
2017/12/14 Python
详解tensorflow实现迁移学习实例
2018/02/10 Python
我喜欢你 抖音表白程序python版
2019/04/07 Python
python 计算数据偏差和峰度的方法
2019/06/29 Python
python命名空间(namespace)简单介绍
2019/08/10 Python
Python中无限循环需要什么条件
2020/05/27 Python
如何使用css3实现一个类在线直播的队列动画的示例代码
2020/06/17 HTML / CSS
台湾团购、宅配和优惠券:17Life
2017/08/14 全球购物
YSL圣罗兰美妆英国官网:Yves Saint Laurent Beauty UK
2019/08/03 全球购物
儿科主治医生个人求职信
2013/09/23 职场文书
面试后感谢信
2014/02/01 职场文书
致100米运动员广播稿
2014/02/14 职场文书
《藤野先生》教学反思
2014/02/19 职场文书
读书活动实施方案
2014/03/10 职场文书
中学生勤俭节约倡议书
2015/04/29 职场文书