更新时间:2024-05-31 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。