深度学习 AMI
开发人员指南
AWS 服务或AWS文档中描述的功能,可能因地区/位置而异。请点击 Amazon AWS 入门,可查看中国地区的具体差异

PyTorch

要激活框架,请按照 采用 Conda 的 Deep Learning AMI 上的说明进行操作。

对于使用 CUDA 8 的 Python 3:

$ source activate pytorch_p36

对于使用 CUDA 8 的 Python 2:

$ source activate pytorch_p27

启动 iPython 终端。

(pytorch_p36)$ ipython

运行快速 PyTorch 程序。

import torch x = torch.rand(5, 3) print(x) print(x.size()) y = torch.rand(5, 3) print(torch.add(x, y))

您应该会看到系统输出初始随机数组,然后输出大小,然后添加另一个随机数组。

更多教程

您可以在 DLAMI 主目录的 采用 Conda 的 Deep Learning AMI 教程文件夹中找到更多教程。如需查看更多教程和示例,请参阅该框架的官方文档、PyTorch 文档PyTorch 网站。