PyTorch

conda install pytorch torchvision torchaudio cpuonly -c pytorch

python -m pip install --upgrade pip
python -m pip install --upgrade pip  -i https://pypi.douban.com/simple/

conda create -n ptenv python=3.6

PyTorch安装‘

import torch
from torch import autograd

x = torch.tensor(1.)
a = torch.tensor(1., requires_grad=True)
b = torch.tensor(2., requires_grad=True)
c = torch.tensor(3., requires_grad=True)

y = a**2 * x + b * x + c

print('before', a.grad, b.grad, c.grad)
grads = autograd.grad(y, [a, b, c])
print('after:', grads[0], grads[1], grads[2])