pytorch开发环境搭建(win10)

一、Pytorch简介

        Pytorch是2017年Facebook人工智能研究院团队在Github上开源的一个深度学习框架,其历史可追溯到2002年诞生于纽约大学的Torch。自发布之后,Pytorch的热度便持续高涨,随着深度学习持续火热,Pytorch这一深度学习框架被运用地越发广泛。



二、环境搭建

      1、Anaconda安装

        pytorch开发,推荐使用Anaconda,因为集成了很多相关的工具包,在Anaconda官网选择自己电脑的环境下载相关安装包即可安装。

      2、 Pytorch安装

   进入Pytorch的官网,根据自己的电脑环境配置选择,即可得到相关的安装的命令,之后在Anaconda Prompt中输入命令即可完成Pytorch的安装(cuda是有GPU的时候选择,使用CPU的话选择None即可)。

  3 、进入命令窗口

  执行   

conda install pytorch torchvision cpuonly -c pytorch

     

等待安装完成,中途可能由于网络原因,报些HTTP错误,这个只要重新执行一下命令就行

三、验证torch安装

    cmd进入窗口,

  输入 python 

再次输入

      import torch

     Torch=torch.rand(5,5)

      print(Torch)

说明pytorch环境安装成功

发布了57 篇原创文章 · 获赞 22 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/yb546822612/article/details/103664182