更新时间:2024-12-17 GMT+08:00
分享

Caffe分类范例

本实践采用caffe官方的分类例子,地址为https://github.com/BVLC/caffe/blob/master/examples/00-classification.ipynb

使用CPU

创建一个普通job,镜像输入第三方镜像bvlc/caffe:cpu,设置对应的容器规格。

启动命令添加python /home/caffeEx00.py。

挂载刚刚导入的OBS存储盘:

单击“创建”。等待job执行完成,进入OBS存储盘的outputimg下,可以看到推理使用的图片。

登录在集群中添加GPU节点添加的节点,执行docker logs {容器id}查看归类结果,可以看到结果:tabby cat。

使用GPU

创建一个普通job,镜像输入第三方镜像bvlc/caffe:gpu,设置对应的容器规格。

启动命令添加python /home/caffeEx00_GPU.py。

挂载刚刚导入的OBS存储盘:

单击“创建”。等待job执行完成,进入OBS存储盘的outputimg下,可以看到推理使用的图片。

登录在集群中添加GPU节点添加的节点,执行docker logs {容器id}查看归类结果,可以看到结果:tabby cat。

相关文档