Add PyTorch 1.13.1 and 2.0.1 images

Author: anibali

.github/workflows/publish_1.11.0-cuda11.5-ubuntu20.04.yml

@@ -1,12 +1,12 @@
-name: Publish anibali/pytorch:1.10.2-cuda11.3-ubuntu20.04
+name: Publish anibali/pytorch:1.13.1-cuda11.7-ubuntu22.04
 
 on:
   push:
     branches:
     - master
     paths:
-    - dockerfiles/1.10.2-cuda11.3-ubuntu20.04/*
-    - .github/workflows/publish_1.10.2-cuda11.3-ubuntu20.04.yml
+    - dockerfiles/1.13.1-cuda11.7-ubuntu22.04/*
+    - .github/workflows/publish_1.13.1-cuda11.7-ubuntu22.04.yml
   workflow_dispatch:
 
 jobs:
@@ -19,5 +19,5 @@ jobs:
         username: ${{ secrets.DOCKER_USERNAME }}
         password: ${{ secrets.DOCKER_PASSWORD }}
         repository: anibali/pytorch
-        tags: 1.10.2-cuda11.3-ubuntu20.04,1.10.2-cuda11.3
-        path: dockerfiles/1.10.2-cuda11.3-ubuntu20.04
+        tags: 1.13.1-cuda11.7-ubuntu22.04,1.13.1-cuda11.7
+        path: dockerfiles/1.13.1-cuda11.7-ubuntu22.04

.github/workflows/publish_1.11.0-nocuda-ubuntu20.04.yml

@@ -1,12 +1,12 @@
-name: Publish anibali/pytorch:1.13.0-nocuda-ubuntu22.04
+name: Publish anibali/pytorch:1.13.1-nocuda-ubuntu22.04
 
 on:
   push:
     branches:
     - master
     paths:
-    - dockerfiles/1.13.0-nocuda-ubuntu22.04/*
-    - .github/workflows/publish_1.13.0-nocuda-ubuntu22.04.yml
+    - dockerfiles/1.13.1-nocuda-ubuntu22.04/*
+    - .github/workflows/publish_1.13.1-nocuda-ubuntu22.04.yml
   workflow_dispatch:
 
 jobs:
@@ -19,5 +19,5 @@ jobs:
         username: ${{ secrets.DOCKER_USERNAME }}
         password: ${{ secrets.DOCKER_PASSWORD }}
         repository: anibali/pytorch
-        tags: 1.13.0-nocuda-ubuntu22.04,1.13.0-nocuda
-        path: dockerfiles/1.13.0-nocuda-ubuntu22.04
+        tags: 1.13.1-nocuda-ubuntu22.04,1.13.1-nocuda
+        path: dockerfiles/1.13.1-nocuda-ubuntu22.04

.github/workflows/publish_1.13.0-cuda11.8-ubuntu22.04.yml

@@ -1,12 +1,12 @@
-name: Publish anibali/pytorch:1.10.2-nocuda-ubuntu20.04
+name: Publish anibali/pytorch:2.0.1-cuda11.8-ubuntu22.04
 
 on:
   push:
     branches:
     - master
     paths:
-    - dockerfiles/1.10.2-nocuda-ubuntu20.04/*
-    - .github/workflows/publish_1.10.2-nocuda-ubuntu20.04.yml
+    - dockerfiles/2.0.1-cuda11.8-ubuntu22.04/*
+    - .github/workflows/publish_2.0.1-cuda11.8-ubuntu22.04.yml
   workflow_dispatch:
 
 jobs:
@@ -19,5 +19,5 @@ jobs:
         username: ${{ secrets.DOCKER_USERNAME }}
         password: ${{ secrets.DOCKER_PASSWORD }}
         repository: anibali/pytorch
-        tags: 1.10.2-nocuda-ubuntu20.04,1.10.2-nocuda
-        path: dockerfiles/1.10.2-nocuda-ubuntu20.04
+        tags: 2.0.1-cuda11.8-ubuntu22.04,2.0.1-cuda11.8,latest
+        path: dockerfiles/2.0.1-cuda11.8-ubuntu22.04

.github/workflows/publish_1.10.2-cuda11.3-ubuntu20.04.yml .github/workflows/publish_1.13.1-cuda11.7-ubuntu22.04.yml

@@ -1,12 +1,12 @@
-name: Publish anibali/pytorch:2.0.0-nocuda-ubuntu22.04
+name: Publish anibali/pytorch:2.0.1-nocuda-ubuntu22.04
 
 on:
   push:
     branches:
     - master
     paths:
-    - dockerfiles/2.0.0-nocuda-ubuntu22.04/*
-    - .github/workflows/publish_2.0.0-nocuda-ubuntu22.04.yml
+    - dockerfiles/2.0.1-nocuda-ubuntu22.04/*
+    - .github/workflows/publish_2.0.1-nocuda-ubuntu22.04.yml
   workflow_dispatch:
 
 jobs:
@@ -19,5 +19,5 @@ jobs:
         username: ${{ secrets.DOCKER_USERNAME }}
         password: ${{ secrets.DOCKER_PASSWORD }}
         repository: anibali/pytorch
-        tags: 2.0.0-nocuda-ubuntu22.04,2.0.0-nocuda
-        path: dockerfiles/2.0.0-nocuda-ubuntu22.04
+        tags: 2.0.1-nocuda-ubuntu22.04,2.0.1-nocuda
+        path: dockerfiles/2.0.1-nocuda-ubuntu22.04

.github/workflows/publish_1.13.0-nocuda-ubuntu22.04.yml .github/workflows/publish_1.13.1-nocuda-ubuntu22.04.yml

@@ -37,10 +37,10 @@ access within Docker containers. This can be found at
 Prebuilt images are available on Docker Hub under the name
 [anibali/pytorch](https://hub.docker.com/r/anibali/pytorch/).
 
-For example, you can pull an image with PyTorch 2.0.0 and CUDA 11.8 using:
+For example, you can pull an image with PyTorch 2.0.1 and CUDA 11.8 using:
 
 ```bash
-$ docker pull anibali/pytorch:2.0.0-cuda11.8
+$ docker pull anibali/pytorch:2.0.1-cuda11.8
 ```
 
 
@@ -109,12 +109,12 @@ The recommended way of adding additional dependencies to an image is to create
 your own Dockerfile using one of the PyTorch images from this project as a base.
 
 For example, let's say that you require OpenCV and wish to work with PyTorch
-2.0.0. You can create your own Dockerfile using
-`anibali/pytorch:2.0.0-cuda11.8-ubuntu22.04` as the base image and install
+2.0.1. You can create your own Dockerfile using
+`anibali/pytorch:2.0.1-cuda11.8-ubuntu22.04` as the base image and install
 OpenCV using additional build steps:
 
 ```dockerfile
-FROM anibali/pytorch:2.0.0-cuda11.8-ubuntu22.04
+FROM anibali/pytorch:2.0.1-cuda11.8-ubuntu22.04
 
 # Set up time zone.
 ENV TZ=UTC

.github/workflows/publish_2.0.0-cuda11.8-ubuntu22.04.yml

@@ -0,0 +1,47 @@
+FROM nvidia/cuda:11.7.1-base-ubuntu22.04
+
+# Remove any third-party apt sources to avoid issues with expiring keys.
+RUN rm -f /etc/apt/sources.list.d/*.list
+
+# Install some basic utilities.
+RUN apt-get update && apt-get install -y \
+    curl \
+    ca-certificates \
+    sudo \
+    git \
+    bzip2 \
+    libx11-6 \
+ && rm -rf /var/lib/apt/lists/*
+
+# Create a working directory.
+RUN mkdir /app
+WORKDIR /app
+
+# Create a non-root user and switch to it.
+RUN adduser --disabled-password --gecos '' --shell /bin/bash user \
+ && chown -R user:user /app
+RUN echo "user ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/90-user
+USER user
+
+# All users can use /home/user as their home directory.
+ENV HOME=/home/user
+RUN mkdir $HOME/.cache $HOME/.config \
+ && chmod -R 777 $HOME
+
+# Download and install Micromamba.
+RUN curl -sL https://micro.mamba.pm/api/micromamba/linux-64/1.1.0 \
+  | sudo tar -xvj -C /usr/local bin/micromamba
+ENV MAMBA_EXE=/usr/local/bin/micromamba \
+    MAMBA_ROOT_PREFIX=/home/user/micromamba \
+    CONDA_PREFIX=/home/user/micromamba \
+    PATH=/home/user/micromamba/bin:$PATH
+
+# Set up the base Conda environment by installing PyTorch and friends.
+COPY conda-linux-64.lock /app/conda-linux-64.lock
+RUN micromamba create -qy -n base -f /app/conda-linux-64.lock \
+ && rm /app/conda-linux-64.lock \
+ && micromamba shell init --shell=bash --prefix="$MAMBA_ROOT_PREFIX" \
+ && micromamba clean -qya
+
+# Set the default command to python3.
+CMD ["python3"]