diff --git a/.github/workflows/deploy_images.yml b/.github/workflows/deploy_images.yml index 9bb323d..b34f3a1 100644 --- a/.github/workflows/deploy_images.yml +++ b/.github/workflows/deploy_images.yml @@ -127,7 +127,26 @@ jobs: push: ${{ github.event.inputs.skip_push != 'true' }} username: ${{ secrets.DOCKERHUB_USERNAME }} password: ${{ secrets.DOCKERHUB_TOKEN }} + + py311-cuda: + runs-on: ubuntu-latest + steps: + - name: Checkout 🛎️ + uses: actions/checkout@v3 + with: + persist-credentials: false + - name: Build and deploy image 🐳 + uses: docker/build-push-action@v1 + with: + path: docker + dockerfile: docker/Dockerfile_311_cuda + repository: cmsml/cmsml + tags: "3.11" + push: ${{ github.event.inputs.skip_push != 'true' }} + username: ${{ secrets.DOCKERHUB_USERNAME }} + password: ${{ secrets.DOCKERHUB_TOKEN }} + docs: needs: py39 runs-on: ubuntu-latest diff --git a/docker/Dockerfile_311_cuda b/docker/Dockerfile_311_cuda new file mode 100644 index 0000000..8315a38 --- /dev/null +++ b/docker/Dockerfile_311_cuda @@ -0,0 +1,38 @@ +FROM python:3.11 + +# set the workdir +WORKDIR /root + +# minimal software stack +RUN apt-get update; apt-get clean +RUN apt-get install -y nano less htop git; apt-get clean + +# python software stack +RUN pip install --no-cache-dir --upgrade pip setuptools +RUN pip install --no-cache-dir --upgrade ipython +RUN pip install --no-cache-dir \ + numpy \ + scipy \ + matplotlib \ + pandas \ + numexpr \ + jupyterlab \ + notebook \ + scikit-learn \ + scikit-optimize \ + tensorflow[and-cuda] \ + xgboost \ + scinum \ + nvidia_smi \ + py3nvml \ + torch + +# install cmsml from master +RUN git clone https://github.com/cms-ml/cmsml.git && \ + cd cmsml && \ + pip install --no-cache-dir .[dev] && \ + cd .. && \ + rm -rf cmsml + +# initial command +CMD ["bash", "-i", "-l"]