1.官网进入 https://pytorch.org/get-started/locally/
选择自己相应的版本,若安装GPU版,选择相应的CUDA版本,若只是安装CPU版的,请在CUDA一项上选择None。
2.由于下载速度较慢,这里选择国内镜像安装
只需要在原命令的后面加上
-i https://pypi.tuna.tsinghua.edu.cn/simple some-package
即可。
此时命令变为下图:
(若安装GPU版,需要安装CUDA,可参考博主另一篇文 https://www.jianshu.com/p/b9fea6f3d973 )
3.检查是否安装成功
(这里采用的是jupyter,任意python的环境下均可,输入两行代码后能正确输出版本号即可。)