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
未完待续