update images

puzl-cloud · Oct 16, 2023 · ed62a75 · ed62a75
1 parent 6a69332
commit ed62a75
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 15 deletions.
diff --git a/images/caffe2/cuda11-8/Dockerfile b/images/caffe2/cuda11-8/Dockerfile
@@ -5,7 +5,7 @@ ARG INTERPRETER
 ARG INTERPRETER_VERSION
 
 LABEL [email protected] \
-      vendor=puzl.ee
+      vendor=https://puzl.cloud
 
 ENV USER=ubuntu \
     UID=1000 \
@@ -102,14 +102,25 @@ RUN rm -rf /usr/local/bin/pip3 \
 
 RUN ${INTERPRETER} -m pip install -r /tmp/requirements.txt
 
-RUN ${INTERPRETER} -m pip install \
+RUN case ${INTERPRETER} in \
+    python3.9) \
+      export CP_FORMAT=cp39 \
+      ;; \
+    python3.10) \
+      export CP_FORMAT=cp310 \
+      ;; \
+    python3.11) \
+      export CP_FORMAT=cp311 \
+      ;; \
+  esac \
+  && ${INTERPRETER} -m pip install \
         torch==2.1.0 \
         torchvision==0.16.0 \
         torchaudio==2.1.0 \
         jupyterlab==4.0.7 \
-        --index-url https://download.pytorch.org/whl/cu118 \
-        --extra-index-url https://pypi.org/simple \
-    && ${INTERPRETER} -m pip cache purge
+      --index-url https://download.pytorch.org/whl/cu118 \
+      --extra-index-url https://pypi.org/simple \
+  && ${INTERPRETER} -m pip cache purge
 
 RUN curl https://rclone.org/install.sh -o ~/install.sh \
     && chmod +x ~/install.sh \

diff --git a/images/fastai/cuda11-8/Dockerfile b/images/fastai/cuda11-8/Dockerfile
@@ -5,7 +5,7 @@ ARG INTERPRETER
 ARG INTERPRETER_VERSION
 
 LABEL [email protected] \
-      vendor=puzl.ee
+      vendor=https://puzl.cloud
 
 ENV USER=ubuntu \
     UID=1000 \
@@ -102,15 +102,26 @@ RUN rm -rf /usr/local/bin/pip3 \
 
 RUN ${INTERPRETER} -m pip install -r /tmp/requirements.txt
 
-RUN ${INTERPRETER} -m pip install \
+RUN case ${INTERPRETER} in \
+    python3.9) \
+      export CP_FORMAT=cp39 \
+      ;; \
+    python3.10) \
+      export CP_FORMAT=cp310 \
+      ;; \
+    python3.11) \
+      export CP_FORMAT=cp311 \
+      ;; \
+  esac \
+  && ${INTERPRETER} -m pip install \
         torch==2.0.1 \
         torchvision==0.15.2 \
         torchaudio==2.0.2 \
         fastai==2.7.13 \
         jupyterlab==4.0.7 \
-        --index-url https://download.pytorch.org/whl/cu118 \
-        --extra-index-url https://pypi.org/simple \
-    && ${INTERPRETER} -m pip cache purge
+      --index-url https://download.pytorch.org/whl/cu118 \
+      --extra-index-url https://pypi.org/simple \
+  && ${INTERPRETER} -m pip cache purge
 
 RUN curl https://rclone.org/install.sh -o ~/install.sh \
     && chmod +x ~/install.sh \

diff --git a/images/keras/cuda11-8/Dockerfile b/images/keras/cuda11-8/Dockerfile
@@ -109,16 +109,15 @@ RUN rm -rf /usr/local/bin/pip3 \
 
 RUN ${INTERPRETER} -m pip install -r /tmp/requirements.txt
 
-RUN \
-  case ${INTERPRETER} in \
+RUN case ${INTERPRETER} in \
     python3.9) \
-      CP_FORMAT=cp39 \
+      export CP_FORMAT=cp39 \
       ;; \
     python3.10) \
-      CP_FORMAT=cp310 \
+      export CP_FORMAT=cp310 \
       ;; \
     python3.11) \
-      CP_FORMAT=cp311 \
+      export CP_FORMAT=cp311 \
       ;; \
   esac \
   && ${INTERPRETER} -m pip install \