windows pytorch 环境搭建

参考:https://www.cnblogs.com/GrPhoenix/p/10324172.html

1.Anaconda已经安装完毕

2.创建python虚拟环境:conda create -n pytorch-gpu python=3.6

3.激活创建的虚拟环境:activate pytorch-gpu

4.安装pytorch:  https://pytorch.org/get-started/previous-versions/  :conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

5.如果安装太慢,可以先下载到本地:torch-1.2.0-cp36-cp36m-win_amd64.whl

6.安装pip install torch-1.2.0-cp36-cp36m-win_amd64.whl

7.测试:

>>> import torch
>>> torch.cuda.is_available()

返回true,这GPU版本的可用。

原文地址:https://www.cnblogs.com/shellycn/p/13071102.html