search

NVlabs / mask-auto-labeler

Other
160 stars 13 forks source link

Docker update and maintenance #12

Open tianyufang1958 opened 1 year ago

tianyufang1958 commented 1 year ago

The current docker file is based on old versions of MMCV and pytorch, and does work at the moment. Would you plan to maintain and update the code also docker file to meet the new versions of dependencies?

voidrank commented 1 year ago

Hi @tianyufang1958 ,

Thanks for the feedback!

No, I don’t have a plan to update the docker. If you have a stable and later docker image/file, could you share and let me put them on the readme?

Shiyi

tianyufang1958 commented 1 year ago

Hi @tianyufang1958 ,

Thanks for the feedback!

No, I don’t have a plan to update the docker. If you have a stable and later docker image/file, could you share and let me put them on the readme?

Shiyi

Sure I will test and update with you.

voidrank commented 1 year ago

Hi @tianyufang1958 , do you have any update?

tianyufang1958 commented 1 year ago

Hi @tianyufang1958 , do you have any update?

Hi, Yes I will try to update before the end of the week. Sorry for the slow response.

tianyufang1958 commented 1 year ago

Hi @tianyufang1958 , do you have any update?

Please try this one.

FROM nvcr.io/nvidia/pytorch:21.12-py3 RUN apt-get update RUN apt-get install -y htop vim tmux gcc g++ psmisc iputils-ping RUN apt-get install -y libgl1 zip RUN pip install opencv-python==4.5.5.64 RUN pip install scipy==1.6.3 pytorch_lightning==1.8.1 torchmetrics==0.10.2 mmcv-full==1.7.0 mmdet==2.25.3 mmcls==0.24.1 RUN pip install einops==0.6.0 shapely RUN pip install timm RUN pip install git+https://github.com/lvis-dataset/lvis-api.git RUN python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'