深度学习 AMI
开发人员指南
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 Amazon AWS 入门

PyTorch

PyTorch 教程

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

对于使用 CUDA 9 和 cuDNN 7 的 Python 3:

$ source activate pytorch_p36

对于使用 CUDA 9 和 cuDNN 7 的 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 网站。

本页内容: