环境准备
Anaconda3 (with Python 3.5/3.6)
Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
CUDA 8 / CUDA 9(包中已内置,因此老版本也能用)
cuDNN v5以上(包中已经自带了, cuDNN v5用户亦可放心使用)
如果不想放弃Anaconda2也可以创建虚拟环境来使用Anaconda3。
Anaconda2与Anaconda3共存方法,详细参考:http://blog.youkuaiyun.com/infin1te/article/details/50445217
安装好Anaconda3后activate Anaconda3,再conda create –name pytorch,为pytorch创建一个虚拟环境。
Pytorch安装
激活pytorch
activate Anaconda3
activate pytorch
由于是在Anaconda2下的Anaconda3中新建的虚拟环境,所以每次激活pytorch时,需要先激活Anaconda3。安装依赖项