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、个人思考
-
因为之前没有了解过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
-
下载了之后,它该怎样用呢。找了个小例子,用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。
-
pytorch作为深度学习框架,它与tensorflow相比较有什么有缺点呢?
参考tensorflow和pytorch框架对比总结
https://blog.csdn.net/WJ_MeiMei/article/details/88724984
来源:CSDN
作者:酸梅果茶
链接:https://blog.csdn.net/qq_41821678/article/details/104642905