Windows+Anaconda+Pytorch安装教程(CPU版本)

旧巷老猫 提交于 2020-03-05 01:28:20

Windows+Anaconda+Pytorch安装教程(CPU版本)

1、教程背景

最近由于要听某节公开课,老师用的是Pytorch,故不得不装Pytorch。花费了半天时间,终于安装好了。网上的教程很多不适用(也可能是我没弄对),安装过程中还是踩了很多坑。

本文将自己安装Pytorch的心得记录下来,不一定适用你的电脑环境,但是可以知道安装过程中应该注意哪些要点。

如果有侵权或疑问,可以联系 zhankun3280@139.com

2、安装过程

我的电脑是Win10系统,CPU,已装过python3.5,没有装过 Anaconda。

步骤1: 安装Anaconda。官网https://www.anaconda.com/distribution/,我下载的是Python 3.7 version、64-Bit(PS:我的电脑装本来装有python3.5,暂时不知道再装个3.7版本的有没有影响,有影响后续再更新说明。)

在这里插入图片描述
具体安装步骤不再详述,一步一步来就行,网上也有很多教程。

步骤2: 安装完Anaconda后,找到Anaconda Prompt(Anaconda)并打开。
在这里插入图片描述
步骤3: 安装pytorch。(网上有很多命令方法实现,我试了很多,几乎没有能走的通的,很多都是下载不了,最常见的问题就是:网络不好 ,每次下载一半就出现Read timed out。)

后来查资料,发现直接在Anaconda里面下载挺好用的。用conda安装(不是pip)。
安装过程中,如果网速不好,可能很费时间,需要多次安装,也有可能出现Read timed out错误。

conda install pytorch-cpu  #后面没有 -c pytorch

在这里插入图片描述
步骤4: 安装完,测试是否安装成功。直接python运行,import torch,如果没有报错,则安装成功。
在这里插入图片描述
注意: 我用网上的教程总是出现以下问题,而且查找了很多方法也没有解决,这才放弃,采用Anaconda安装。
在这里插入图片描述
步骤5: 安装torchvision。安装这个一定要注意版本,稍有不慎就会报错。
例如步骤4安装的torch版本是1.1.0,这里如果torchvision安装版本是0.5.0,就会报错,要求torch版本必须是1.4.0。(我就是这样折腾了好久,网速还不好,好不容易下载好了,却发现不能用,特别崩溃~~~)
在这里插入图片描述
所以我下载的torchvision指定版本为0.2.1

pip install torchvision==0.2.1

下载完毕之后进行验证测试,查看安装路径、相关版本。没有报错,即安装成功。

import torch
import torchvision
print(torch)              
print(torchvision)          #查看安装路径
print(torch.__version__)    #查看版本
print(torchvision.__version__)

在这里插入图片描述

3、个人思考

  1. 因为之前没有了解过pytorch,我在下载过程中,发现有时候是torch,有时候又说pytorch,不是很理解他们的关系。后来查资料,大概明白了一些,说白了,pytorch就是torch的python版。

    附上pytorch和torch框架对比(区别 联系)
    https://blog.csdn.net/WJ_MeiMei/article/details/88720146
    附上浅谈Pytorch与Torch的关系
    https://m.oldpan.me/archives/pytorch-torch-relation

  2. 下载了之后,它该怎样用呢。找了个小例子,用pytorch中的随机数生成,画散点图。终端运行test_torch.py

import torch
import numpy as np
import matplotlib.pyplot as plt

x = torch.linspace(-1,1,10).unsqueeze(1) 
y = 3 * x + torch.rand(x.size())      
x = np.array(x)       #x与y为tensor型,需转为numpy才能被画图包plt所用
y = np.array(y)
plt.scatter(x, y)     #画(x,y)的散点图
plt.show()

生成的散点图如下图Figure1。
在这里插入图片描述

  1. pytorch作为深度学习框架,它与tensorflow相比较有什么有缺点呢?

    参考tensorflow和pytorch框架对比总结
    https://blog.csdn.net/WJ_MeiMei/article/details/88724984
    在这里插入图片描述

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!