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

前置环境:

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