win7下一键安装PyTorch

本文参考了:https://zhuanlan.zhihu.com/p/26871672

非常感谢知乎大佬们提供的工具。


首先要满足:

  • Anaconda3 (with Python 3.5/3.6)
  • Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
  • CUDA 8 / CUDA 9(包中已内置,因此老版本也能用)
  • cuDNN v5以上(包中已经自带了, cuDNN v5用户亦可放心使用)
step1. 打开Anaconda Prompt

在开始菜单下找到anaconda所在的文件夹,打开后那个黑色图标就是。

step2. 在Anaconda Prompt下根据自己电脑的版本输入如下命令:

# for CPU only packages
conda install -c peterjc123 pytorch

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80

# for Windows 10 and Windows Server 2016, CUDA 9
conda install -c peterjc123 pytorch cuda90

# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8
conda install -c peterjc123 pytorch_legacy cuda80
剩下的时间,就干点儿别的,一会儿就发现大功告成啦。

你可能感兴趣的:(Python)