Windows下PyTorch简易安装教程

1.安装环境

OS: Windows Server 2016 Standard Evaluation
CUDA: Cuda compilation tools, release 10.1, V10.1.105
Python: Python 3.7.3
conda: conda 4.6.13

因为是在工作站上安装,所以系统是Windows Server的,Windows10的安装方法也一样。
其次,预先安装了Anaconda,所以是Python 3.7 + conda 4.6.13的环境。
建议大家也安装Anaconda,安装简单便捷,反观编译Python3源码实在是太多坑。


2.安装pyorch

确定环境后进入pytorch官网:https://pytorch.org/get-started/locally/
Windows下PyTorch简易安装教程_第1张图片
按环境依次选择后,下方会出现一条命令,提示Run this Command.
打开Anacoda Prompt,输入上图红框内的命令:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

Windows下PyTorch简易安装教程_第2张图片
输入y,继续完成安装即可:

Windows下PyTorch简易安装教程_第3张图片

3.安装验证

打开控制台,可以import这两个模块即表示安装成功。

(base) C:\Users\Administrator>python
Python 3.7.3 (default, Mar 27 2019, 17:13:21) [MSC v.1915 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>>

你可能感兴趣的:(深度学习)