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下搭建安装

相关文章:

  • 2021-04-16
  • 2021-04-23
  • 2021-10-28
  • 2021-09-06
  • 2021-12-12
  • 2021-05-25
  • 2021-09-30
  • 2021-07-03
猜你喜欢
  • 2021-08-04
  • 2022-12-23
  • 2021-07-12
  • 2021-10-05
  • 2021-06-20
  • 2021-10-26
  • 2021-09-24
相关资源
相似解决方案