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

Caffe2

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

只有使用 CUDA 9 的 Python 2 选项:

$ source activate caffe2_p27

启动 iPython 终端。

(caffe2_p27)$ ipython

运行快速 Caffe2 程序。

from caffe2.python import workspace, model_helper import numpy as np # Create random tensor of three dimensions x = np.random.rand(4, 3, 2) print(x) print(x.shape) workspace.FeedBlob("my_x", x) x2 = workspace.FetchBlob("my_x") print(x2)

您应该会看到系统输出初始 numpy 随机数组,然后这些数组加载到了 Caffe2 blob 中。请注意,加载之后,它们是相同的。

更多教程

如需查看更多教程和示例,请参阅该框架的官方 Python 文档、Python API for Caffe2Caffe2 网站。