【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 进行授权