pytorch pip 安装_pytorch windows版安装使用

前段时间入门deep learning，用lenet-5实现了MNIST数据集的分类，最近开始用pytorch正式开始DL。

废话就不多说了进入正题。

利用pip安装pytorch

在安装pytorch前需要先完成以下准备工作

• 安装python

Python Releases for Windows​www.python.org

一般下载了python就已经自带pip，若没有再安装pip

• 安装pip

pip​pypi.org

• 下载numpy库

在cmd中输入

pip3

开始安装pytorch

pytorch官网：https://pytorch.org/get-started/locally/

速度快深度

选择自己电脑对应的版本，stable是稳定版，Package选Pip

查看下载的python版本

python -V

CUDA是根据各自的显卡来判断的，集成显卡，amd一般都是没有的就选None就好了，

如果是NVIDIA的可以打开显卡控制面板

点击帮助里的系统信息查询

CUDA9.2是我的版本

最后把Run this Command框内的内容复制到命令行，安装速度还是很快的。

验证

用管理员身份打开命令行，输入：

python

接着输入：

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

输出（示例）：

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

判断CUDA能否使用

import torch
torch.cuda.is_available()

大功告成！