pytorch 安装指南

1. windows 下

1.安装python环境

conda create -n pytorch1 python=3.6

2.安装pytorch

conda install -n pytorch1 torchvision cudatoolkit=10.1 -c pytorch

3.测试

import torch
print(torch.cuda.is_available())