PyTorch 在Anaconda Windows下搭建安装

1. 登录官网
https://pytorch.org/get-started/locally/#anaconda
PyTorch 在Anaconda Windows下搭建安装
2. 选择pip方式安装
用conda方式安装,缺失很多依赖的包。就直接用pip方式安装了。
pip install torch1.6.0+cu101 torchvision0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

PyTorch 在Anaconda Windows下搭建安装
3. 测试安装是否成功

from future import print_function
import torch
x = torch.rand(5,3)
print(x)
tensor([[0.7961, 0.4219, 0.3522],
[0.0404, 0.7696, 0.5884],
[0.5852, 0.3819, 0.1786],
[0.6307, 0.9079, 0.9156],
[0.4740, 0.4477, 0.1095]])

PyTorch 在Anaconda Windows下搭建安装