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 \