Windows 10系统在Anaconda下安装GPU版Pytorch

时间:2021-06-14 17:21:22   收藏:0   阅读:12

前置环境:

windows10+anaconda3+python3.7+cuda9.1.104+cudnn7.4.2

技术分享图片

受限于Cuda版本,暂时安装pytorch v1.1.0

语句命令为:conda install pytorch==1.1.0 torchvision==0.3.0 -c pytorch

测试:

技术分享图片

 

 

安装完毕

 

原文:https://www.cnblogs.com/xihongqing/p/14882190.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!