在线安装官网网址:https://pytorch.org/
离线下载安装包路径:
https://download.pytorch.org/whl/torch_stable.html
注意区分CPU版本和cuda版本
cuda版本需要提前查询安装的cuda版本,Windows和Linux版本查询方式不同;
选择python版本,和anaconda版本有关;
最后选择需要的torch版本,可选择稳定版
安装时,pip install ***.whl即可
卸载时,pip uninstall ***.whl
查看是否安装成功:
python
import torch
torch.__version__
不报错,且输出安装torch对应的版本号
离线安装方式比较快