Skip to content

Commit

Permalink
Merge pull request #50 from getwilds/deepLearning_python
Browse files Browse the repository at this point in the history
adding Dockerfile for Python Deep Learning Libraries
  • Loading branch information
tefirman authored Oct 18, 2024
2 parents 9f70919 + 61ec982 commit a388bac
Show file tree
Hide file tree
Showing 2 changed files with 44 additions and 0 deletions.
22 changes: 22 additions & 0 deletions pythonDL/Dockerfile_1.0
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@

# Using a CUDA base image
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04

# Adding labels for the GitHub Container Registry
LABEL org.opencontainers.image.title="pythonDL"
LABEL org.opencontainers.image.description="Docker image for the use of Deep Learning Python packages in FH DaSL's WILDS"
LABEL org.opencontainers.image.version="1.0"
LABEL org.opencontainers.image.authors="[email protected]"
LABEL org.opencontainers.image.url=https://hutchdatascience.org/
LABEL org.opencontainers.image.documentation=https://getwilds.org/
LABEL org.opencontainers.image.source=https://github.com/getwilds/wilds-docker-library
LABEL org.opencontainers.image.licenses=MIT

# Installing Python
RUN apt-get update \
&& apt-get install -y --no-install-recommends python3=3.10.6-1~22.04.1 python3-pip=22.0.2+dfsg-1ubuntu0.4 \
&& rm -rf /var/lib/apt/lists/*

# Installing necessary Python modules
RUN pip install --no-cache-dir numpy==1.26.4 seaborn==0.13.2 matplotlib==3.9.2 pandas==2.2.3 scikit-learn==1.5.2 \
scipy==1.14.1 optuna==4.0.0 joblib==1.4.2 keras==3.6.0 tensorflow==2.17.0 torch==2.5.0
22 changes: 22 additions & 0 deletions pythonDL/Dockerfile_latest
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@

# Using a CUDA base image
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04

# Adding labels for the GitHub Container Registry
LABEL org.opencontainers.image.title="pythonDL"
LABEL org.opencontainers.image.description="Docker image for the use of Deep Learning Python packages in FH DaSL's WILDS"
LABEL org.opencontainers.image.version="1.0"
LABEL org.opencontainers.image.authors="[email protected]"
LABEL org.opencontainers.image.url=https://hutchdatascience.org/
LABEL org.opencontainers.image.documentation=https://getwilds.org/
LABEL org.opencontainers.image.source=https://github.com/getwilds/wilds-docker-library
LABEL org.opencontainers.image.licenses=MIT

# Installing Python
RUN apt-get update \
&& apt-get install -y --no-install-recommends python3=3.10.6-1~22.04.1 python3-pip=22.0.2+dfsg-1ubuntu0.4 \
&& rm -rf /var/lib/apt/lists/*

# Installing necessary Python modules
RUN pip install --no-cache-dir numpy==1.26.4 seaborn==0.13.2 matplotlib==3.9.2 pandas==2.2.3 scikit-learn==1.5.2 \
scipy==1.14.1 optuna==4.0.0 joblib==1.4.2 keras==3.6.0 tensorflow==2.17.0 torch==2.5.0

0 comments on commit a388bac

Please sign in to comment.