Python PyTorch 启动

PyTorch 是一个基于 Python 的科学计算包，它拥有以下特点：

• 一个强大的 GPU 加速的张量计算（Tensor）库
• 可以构建动态神经网络，支持自动求导
• 大量的工具和库可以用来构建和训练神经网络

安装

安装 PyTorch 可以使用 pip 命令：

pip install torch torchvision

在使用 torchvision 时，需要使用以下 pip 命令安装：

pip install pillow==7.1.0

简单使用

导入 PyTorch 的方式：

import torch

创建一个张量：

x = torch.randn(5, 3)
print(x)

输出：

tensor([[ 0.0463, -0.2742, -0.4145],
        [ 1.2779, -0.2922,  0.5027],
        [ 0.2387, -0.4043,  2.0527],
        [ 0.3031, -1.0613, -0.1850],
        [-0.7083,  1.0013, -0.6310]])

可以从 NumPy 数组中创建一个张量：

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = torch.from_numpy(a)
print(b)

输出：

tensor([[1, 2],
        [3, 4]], dtype=torch.int32)

参考链接

• PyTorch 官方网站
• PyTorch GitHub