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版本的可用。

posted @ 2020-06-09 12:43  shellycn  阅读(555)  评论(0编辑  收藏  举报