赞
踩
**
PyTorch 是一个基于 Python 的科学计算包,主要定位两类人群:
NumPy 的替代品,可以利用 GPU 的性能进行计算。
深度学习研究平台拥有足够的灵活性和速度
我按照教程安装的,没有装CUDA,直接安装的pytroch
CNN数字识别
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets, transforms
from torch.autograd import Variable
batch_size = 64
train_dataset = datasets.MNIST(root=’./data/’,
train=True,
transform=transforms.ToTensor(),
download=True)
test_dataset = datasets.MNIST(root=’./data/’,
train=False,
transform=
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。