文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Python安装Pytorch最新图文教程

2023-05-15 17:37

关注

最近人工智能等多门课需要复现论文,近两年的论文很多都是基于Pytorch环境做的实验,所以,这里总结一下Pytorch的安装教程,做好最快、最简单、最好地完成安装。

本机环境
Win10+1050Ti+Python3.7

1、查看本机的CUDA版本

cmd命令行输入nvidia-smi,在第一行最右边可以看到CUDA的版本号,我的版本是11.1

在这里插入图片描述

2、安装Pytroch

1、点击进入Pytorch官网

在这里插入图片描述

然后选择Get Started,就是如下界面

在这里插入图片描述

2、这里进行Pytorch版本的选择,首先我选择的是Stable稳定版,然后OS是Windows系统,Package包就使用Conda,Language肯定选Python,最后的Compute Platform就根据大家的需求来定了。大家如果想在自己电脑(具有NVIDIA显卡)上用GPU进行深度学习模型的推理,就选CUDA;如果只想调用CPU运行算法模型的推理,就选CPU。

一句话总结:包含CUDA的Pytorch可以在程序中设置调用GPU或CPU运行,不含CUDA的Pytorch只能调用CPU运行程序。

选CPU的同学可以直接看第3步了。我们在第一步已经看过自己的NAVIDA的CUDA版本了,这里我们一定要选择比自己版本低的CUDA。比如,像我的版本是11.1,那么就只能选择10.2,因为11.3对我来说有点高了。
当前官网最新版是CUDA11.3,如果你的CUDA版本比官网推荐的都低,可以点击下面的Previous versions of PyTorch,查看以前的版本。

在这里插入图片描述

3、安装下anaconda深度学习神器,没有安装的同学,请按照这篇教程安装好anaconda,并创建一个python环境:anaconda安装配置教程,python的版本根据你的需求而定,这里我根据复现算法的需求环境,安装的是python3.7。

在这里插入图片描述

4、打开anaconda命令行,先激活需要安装Pytorch的python环境(这里我将python环境命名为pytorch,环境名称随意),复制第二步最下面那段命令行语句,然后回车执行:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

在这里插入图片描述

5、可以看到将要安装的包里面有pytorch和cudatoolkit,我们输入y确认,然后回车

在这里插入图片描述

6、下面就等待这些package安装成功,可以看到Pytorch还是比较大的,所以耐心等待一下。如果下载过程中卡住的同学,直接到torch镜像源下载安装包进行安装,效果相同。

在这里插入图片描述

7、安装完成后,会提示信息done。

在这里插入图片描述

8、输入pip list,查看我们的包是否被安装,这里可以看到torch相关的包都安装了。

在这里插入图片描述

9、我们输入python进入下Python环境,然后输入import torch,如果没有报错说明可以导入成功。

在这里插入图片描述

10、输入torch.cuda.is_available()查看torch是否可以使用显卡,True就代表可以!

在这里插入图片描述

安装好Pytorch环境后,我们就可以来部署运行些深度学习算法了,比如经典的Yolov5,Mask-RCNN、Swin Transformer等,快去体验吧~

CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。开发人员可以使用C语言来为CUDA™架构编写程序,所编写出的程序可以在支持CUDA™的处理器上以超高性能运行。

到此这篇关于Python安装Pytorch最新图文教程的文章就介绍到这了,更多相关Python安装Pytorch内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯