1. 原则
话不啰嗦,原则1: 学东西能看官网教程的优先看官网,切优先英文版。
直接pytorch -> tutorials:
https://pytorch.org/tutorials/beginner/basics/quickstart_tutorial.html
非常好的官方教程
运行环境尝试使用了官网推荐的MS learning (感觉还可以), google的Colab(个人偏好谷歌的产品,奈何国内网络条件不允许);本着方便自己折腾捣鼓的原则,使用了本地的vscode,下载网址如下:
https://code.visualstudio.com/download
安装pytorch:
打开terminal:配置源:
sudo pip3 install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/
安装
pip3 install torch torchvision
创建文件test.py 并粘贴下面内容进去:
import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor
运行
python3 test.py
如未报错,表示安装成功。