pytorch官网地址
系统配置
- win10(1809)
- python3.6
pytorch版本
- stable 1.0
cuda版本
- cuda 10
安装命令
因为没有安装conda,所以采用pip安装
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
测试
import torch
torch.tensor([1,2])
安装完成