文章

【Python】深度学习库

PyTorch。

【Python】深度学习库

PyTorch

PyTorch是一个由Facebook的人工智能研究团队开发的开源深度学习框架。在2016年发布后,PyTorch很快就因其易用性、灵活性和强大的功能而在科研社区中广受欢迎。

官网有安装命令:

1
2
https://pytorch.org/get-started/locally/
torch torchvision torchaudio

导包示例:

1
2
3
4
5
import torch
from torch import nn, optim
from torch.utils.data import Dataset, DataLoader
import torchvision.transforms as transforms
from torch.optim.optimizer import Optimizer
本文由作者按照 CC BY 4.0 进行授权