Mask R-CNN 复现踩坑记

一、制作镜像
1.背景:caffe2并入pytorch下面,clone一直报错。
我的修改:

RUN git clone --branch master --recursive https://github.com/caffe2/caffe2.git
改为了
RUN git clone --branch master --recursive https://github.com/pytorch/pytorch.git

报错如下:


后来请教前辈说是因为网络原因,他说可以通过clone到主机上,然后COPY/ADD进去,这里需要注意的一个点是,如果直接COPY整个pytorch文件夹,默认是把pytorch下面的所有内容复制到你的目的地下面,而非pytorch文件夹本身,所以我们应该写成如下命令:
WORKDIR pytorch
COPY ./pytorch pytorch
这样就实现了pytorch整个文件夹的复制。

但是后面又出现了这样的问题:


报错是说,
CMake Error at proto/CMakeLists.txt:3 (caffe2_protobuf_generate_cpp_py):
  Unknown CMake command "caffe2_protobuf_generate_cpp_py".

所以我决定现在本地编译好,然后COPY进去

尝试了之后发现,不能在本地编译好,否则CMakeLists.txt里面都是本地的东西,所以接下来的做法是将pytorch detectron cocoapi中比较大的下载到host上,随后COPY/ADD进到镜像里面去。

猜你喜欢

转载自blog.csdn.net/weixin_40516558/article/details/79853955