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

前置环境:

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

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

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

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

测试:

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

安装完毕