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

Caffe2

Caffe2 教程

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

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

本页内容: