前提条件:

  • Anaconda3 x64 (with Python 3.5/3.6)
  • Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
  • GPU版本还需要任意版本的 CUDA (包内置了CUDA 8 / 9 的部分主要二进制文件)

只需根据自己的系统键入下面的一条命令即可:(注:仅 0.3.1,以后不再更新)

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。
# for CPU only packages conda install -c peterjc123 pytorch-cpu # for CUDA 8, Windows 10 or Windows Server 2016 conda install -c peterjc123 pytorch # for CUDA 9, Windows 10 or Windows Server 2016 conda install -c peterjc123 pytorch cuda90 # for CUDA 9.1, Windows 10 or Windows Server 2016 conda install -c peterjc123 pytorch cuda91 # for CUDA 8, Windows 7 or Windows Server 2008/2012 conda install -c peterjc123 pytorch cuda80 # for CUDA 9, Windows 7 or Windows Server 2008/2012 conda install -c peterjc123 pytorch cuda90

 

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄