首页 > Windows开发 > 详细

windows pytorch 环境搭建

时间:2020-06-09 13:44:30      阅读:45      评论:0      收藏:0      [点我收藏+]

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

windows pytorch 环境搭建

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!