separate docker images by cuda versions

logivations · Nov 12, 2024 · 16446fb · 16446fb
1 parent 003cac4
commit 16446fb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -34,3 +34,4 @@ RUN pip install -r requirements/build.txt
 RUN pip install --no-cache-dir -e .
 RUN mim install "mmdet>=3.1.0"
 RUN pip install future tensorboard albumentations
+RUN pip install setuptools==59.5.0
diff --git a/docker/Dockerfile_aws b/docker/Dockerfile_aws
@@ -34,3 +34,4 @@ RUN pip install -r requirements/build.txt
 RUN pip install --no-cache-dir -e .
 RUN mim install "mmdet>=3.1.0"
 RUN pip install future tensorboard albumentations
+RUN pip install setuptools==59.5.0