pytorch/torch安装简明教程

LINUX上安装

0.进入官网https://pytorch.org/，按照你的系统以及配置，获取安装命令

1.安装gpu版本的torch、cuda包(用于torch调用gpu加速库)

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
ps: 需要提前安装好cuda驱动，参见
https://blog.csdn.net/u010212101/article/details/104753998

2.安装cpu版本的torch

pip install torch==1.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

-f  指定版本（一般指定官方版本）
-i 指定镜像
== 后面紧跟版本号

Windows上安装

参见：https://zhuanlan.zhihu.com/p/470841101

未完待续