pytorch环境搭建记录

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])

安装完成