参考: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
【说明】:本文章由站长整理发布,文章内容不代表本站观点,如文中有侵权行为,请与本站客服联系(QQ:254677821)!