之前基本上都用Keras，最近越来越多的听说PyTorch，就想着也学习一下，顺便把学习的过程记录下来。在便于以后自己查阅的同时，希望也能帮到有需要的人吧。
本文介绍了我的安装方法以及安装完成后的测试。
1.安装PyTorch
首先，去PyTorch官网查看PyTorch的安装方法。打开页面后，我们看到PyTorch官方给我们提供了自动生成安装命令的选项卡，如下图所示。
image
根据自己的实际情况选择就行，没有支持CUDA的GPU的话，在
CUDA那一项选None即可。系统安装了CUDA，但不知道CUDA版本的话，运行以下命令查看：
cat /usr/local/cuda/version.txt
选定后，得到了我的安装命令：
pip3 install torch torchvision
如果网络慢的话，可以采用清华的镜像，在命令后加上-i https://pypi.tuna.tsinghua.edu.cn/simple some-package即可，即：
pip3 install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
2.测试安装是否成功
PyTorch官网给了测试安装是否成功的例子，在python中执行来进行测试即可：
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
如果输出类似下图的五行三列矩阵，就说明安装成功了。
image
版权声明：本文为博主原创文章，遵循 CC 4.0 BY 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://ywsun.site/articles/24.html
网友评论