Windows 安装Pytorch

一、简介

PyTorch进行神经网络的学习十分有用。这里介绍一下我在Windows-10中安装PyTorch的过程与建议。

二、环境

  • Windows 10
  • 命令行环境
  • Python 3.7.5
  • CUDA 10.2

三、安装Python和pip

  1. 安装python进入官网选定版本并下载，https://www.python.org/downloads/

    不要选最新的3.9.0版本！！！

  2. 安装结束后进入命令行检查pip

    • 一般来说安装python时自带安装了pip
    • 需要配置pip环境变量
      • C:\Environment\Python-3.7.5\Scripts\
      • 这是我自己的安装路径，请将Scripts放进配置环境
    • 需要将pip升级到最新版本
      • python -m pip install --upgrade pip
      • 请在安装torch之前在命令行检查或升级pip

四、进入PyTorch官网，并选择相应的版本

  • 进入官网https://pytorch.org/get-started/locally/
  • 下拉找到这里

  • 建议安装Stable版本
  • 在Run This Command中选中复制文本，粘贴至命令行并回车安装

五、安装完成，测试Torch

  • 进入Torch交互模式
  • 输入以下代码，生成一个3*5的随机矩阵，并输出
from __future__ import print_function
import torch
x=torch.rand(5,3)
print(x)
  • 我的结果如下
tensor([[0.1040, 0.6027, 0.4119],
        [0.2021, 0.4975, 0.2911],
        [0.4338, 0.8703, 0.9059],
        [0.5920, 0.5183, 0.8482],
        [0.2389, 0.2982, 0.7928]])
  • 测试GPU驱动与CUDA的正常，则输入如下代码：
import torch
torch.cuda.is_available()
  • 我的结果如下
true

六、安装建议

  • 不能禾斗上网的上网的朋友们可以尝试换用国内镜像或者将pytorch下载包安装到本地，直接输入Run This Command中的网址，完成后使用pip install安装
  • 慎重使用Pycharm在setting中自动安装

七、参考

随笔参考https://blog.csdn.net/Orange_Spotty_Cat/article/details/90518752