cuda11.1

rocker-org · Jun 1, 2021 · 65f5249 · 65f5249
1 parent aa02167
commit 65f5249
Show file tree

Hide file tree

Showing 10 changed files with 266 additions and 32 deletions.
diff --git a/compose/ml-cuda11.1-4.1.0.yml b/compose/ml-cuda11.1-4.1.0.yml
@@ -0,0 +1,21 @@
+version: '3'
+services:
+  r-ver-4.1.0-cuda11.1:
+    image: rocker/r-ver:4.1.0-cuda11.1
+    build:
+      context: ..
+      dockerfile: dockerfiles/Dockerfile_r-ver_4.1.0-cuda11.1
+  ml-4.1.0-cuda11.1:
+    image: rocker/ml:4.1.0-cuda11.1
+    depends_on:
+    - r-ver-4.1.0-cuda11.1
+    build:
+      context: ..
+      dockerfile: dockerfiles/Dockerfile_ml_4.1.0-cuda11.1
+  ml-verse-4.1.0-cuda11.1:
+    image: rocker/ml-verse:4.1.0-cuda11.1
+    depends_on:
+    - ml-4.1.0-cuda11.1
+    build:
+      context: ..
+      dockerfile: dockerfiles/Dockerfile_ml-verse_4.1.0-cuda11.1
diff --git a/compose/ml-cuda11.1-devel.yml b/compose/ml-cuda11.1-devel.yml
@@ -0,0 +1,21 @@
+version: '3'
+services:
+  r-ver-devel-cuda11.1:
+    image: rocker/r-ver:devel-cuda11.1
+    build:
+      context: ..
+      dockerfile: dockerfiles/Dockerfile_r-ver_devel-cuda11.1
+  ml-devel-cuda11.1:
+    image: rocker/ml:devel-cuda11.1
+    depends_on:
+    - r-ver-devel-cuda11.1
+    build:
+      context: ..
+      dockerfile: dockerfiles/Dockerfile_ml_devel-cuda11.1
+  ml-verse-devel-cuda11.1:
+    image: rocker/ml-verse:devel-cuda11.1
+    depends_on:
+    - ml-devel-cuda11.1
+    build:
+      context: ..
+      dockerfile: dockerfiles/Dockerfile_ml-verse_devel-cuda11.1
diff --git a/dockerfiles/Dockerfile_ml-verse_4.1.0-cuda11.1 b/dockerfiles/Dockerfile_ml-verse_4.1.0-cuda11.1
@@ -1,9 +1,9 @@
 FROM rocker/ml:4.1.0-cuda11.1
 
-LABEL org.label-schema.license="GPL-2.0" \
-      org.label-schema.vcs-url="https://github.com/rocker-org/rocker-versioned" \
-      org.label-schema.vendor="Rocker Project" \
-      maintainer="Carl Boettiger <[email protected]>"
+LABEL org.opencontainers.image.licenses="GPL-2.0-or-later" \
+      org.opencontainers.image.source="https://github.com/rocker-org/rocker-versioned2" \
+      org.opencontainers.image.vendor="Rocker Project" \
+      org.opencontainers.image.authors="Carl Boettiger <[email protected]>"
 
 
 

diff --git a/dockerfiles/Dockerfile_ml-verse_devel-cuda11.1 b/dockerfiles/Dockerfile_ml-verse_devel-cuda11.1
@@ -0,0 +1,16 @@
+FROM rocker/ml:devel-cuda11.1
+
+LABEL org.opencontainers.image.licenses="GPL-2.0-or-later" \
+      org.opencontainers.image.source="https://github.com/rocker-org/rocker-versioned2" \
+      org.opencontainers.image.vendor="Rocker Project" \
+      org.opencontainers.image.authors="Carl Boettiger <[email protected]>"
+
+
+
+RUN /rocker_scripts/install_verse.sh
+RUN /rocker_scripts/install_geospatial.sh
+
+
+
+
+
diff --git a/dockerfiles/Dockerfile_ml_4.1.0-cuda11.1 b/dockerfiles/Dockerfile_ml_4.1.0-cuda11.1
@@ -1,15 +1,12 @@
 FROM rocker/r-ver:4.1.0-cuda11.1
 
-LABEL org.label-schema.license="GPL-2.0" \
-      org.label-schema.vcs-url="https://github.com/rocker-org/rocker-versioned" \
-      org.label-schema.vendor="Rocker Project" \
-      maintainer="Carl Boettiger <[email protected]>"
-
-ENV S6_VERSION=v1.21.7.0
-ENV RSTUDIO_VERSION=latest
-ENV PANDOC_VERSION=default
-ENV TENSORFLOW_VERSION=gpu
-ENV KERAS_VERSION=default
+LABEL org.opencontainers.image.licenses="GPL-2.0-or-later" \
+      org.opencontainers.image.source="https://github.com/rocker-org/rocker-versioned2" \
+      org.opencontainers.image.vendor="Rocker Project" \
+      org.opencontainers.image.authors="Carl Boettiger <[email protected]>"
+
+ENV S6_VERSION=v2.1.0.2
+ENV RSTUDIO_VERSION=preview
 ENV PATH=/usr/lib/rstudio-server/bin:$PATH
 
 

diff --git a/dockerfiles/Dockerfile_ml_devel-cuda11.1 b/dockerfiles/Dockerfile_ml_devel-cuda11.1
@@ -0,0 +1,22 @@
+FROM rocker/r-ver:devel-cuda11.1
+
+LABEL org.opencontainers.image.licenses="GPL-2.0-or-later" \
+      org.opencontainers.image.source="https://github.com/rocker-org/rocker-versioned2" \
+      org.opencontainers.image.vendor="Rocker Project" \
+      org.opencontainers.image.authors="Carl Boettiger <[email protected]>"
+
+ENV S6_VERSION=v2.1.0.2
+ENV RSTUDIO_VERSION=preview
+ENV PATH=/usr/lib/rstudio-server/bin:$PATH
+
+
+RUN /rocker_scripts/install_rstudio.sh
+RUN /rocker_scripts/install_pandoc.sh
+RUN /rocker_scripts/install_tidyverse.sh
+
+EXPOSE 8787
+
+CMD ["/init"]
+
+
+
diff --git a/dockerfiles/Dockerfile_r-ver_4.1.0-cuda11.1 b/dockerfiles/Dockerfile_r-ver_4.1.0-cuda11.1
@@ -1,27 +1,27 @@
-FROM rocker/r-ver:4.1.0
-
-LABEL org.label-schema.license="GPL-2.0" \
-      org.label-schema.vcs-url="https://github.com/rocker-org/rocker-versioned" \
-      org.label-schema.vendor="Rocker Project" \
-      maintainer="Carl Boettiger <[email protected]>"
-
-ENV CUDA_VERSION=11.1
-ENV NCCL_VERSION=2.7.8
-ENV NVIDIA_VISIBLE_DEVICES=all
-ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
-ENV NVIDIA_REQUIRE_CUDA=cuda>=11.1 brand=tesla,driver>=418,driver<419 brand=tesla,driver>=440,driver<441 brand=tesla,driver>=450,driver<451
-ENV CUDA_HOME=/usr/local/cuda
-ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64:$CUDA_HOME/lib64/libnvblas.so:
-ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
+FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
+
+LABEL org.opencontainers.image.licenses="GPL-2.0-or-later" \
+      org.opencontainers.image.source="https://github.com/rocker-org/rocker-versioned2" \
+      org.opencontainers.image.vendor="Rocker Project" \
+      org.opencontainers.image.authors="Carl Boettiger <[email protected]>"
+
+ENV R_VERSION=4.1.0
+ENV TERM=xterm
+ENV LC_ALL=en_US.UTF-8
+ENV LANG=en_US.UTF-8
+ENV R_HOME=/usr/local/lib/R
+ENV CRAN=https://packagemanager.rstudio.com/all/__linux__/focal/latest
+ENV TZ=Etc/UTC
 ENV NVBLAS_CONFIG_FILE=/etc/nvblas.conf
 ENV WORKON_HOME=/opt/venv
 ENV PYTHON_VENV_PATH=/opt/venv/reticulate
 ENV PYTHON_CONFIGURE_OPTS=--enable-shared
-ENV RETICULATE_MINICONDA_ENABLED=FALSE
-ENV PATH=${PYTHON_VENV_PATH}/bin:${CUDA_HOME}/bin:/usr/local/nviida/bin:${PATH}:/usr/local/texlive/bin/x86_64-linux
+ENV RETICULATE_AUTOCONFIGURE=0
+ENV PATH=${PYTHON_VENV_PATH}/bin:${PATH}:${CUDA_HOME}/bin
 
+COPY scripts /rocker_scripts
 
-RUN /rocker_scripts/install_cuda-11.1.sh
+RUN /rocker_scripts/install_R.sh
 RUN /rocker_scripts/config_R_cuda.sh
 RUN /rocker_scripts/install_python.sh
 

diff --git a/dockerfiles/Dockerfile_r-ver_devel-cuda11.1 b/dockerfiles/Dockerfile_r-ver_devel-cuda11.1
@@ -0,0 +1,31 @@
+FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
+
+LABEL org.opencontainers.image.licenses="GPL-2.0-or-later" \
+      org.opencontainers.image.source="https://github.com/rocker-org/rocker-versioned2" \
+      org.opencontainers.image.vendor="Rocker Project" \
+      org.opencontainers.image.authors="Carl Boettiger <[email protected]>"
+
+ENV R_VERSION=devel
+ENV TERM=xterm
+ENV LC_ALL=en_US.UTF-8
+ENV LANG=en_US.UTF-8
+ENV R_HOME=/usr/local/lib/R
+ENV CRAN=https://packagemanager.rstudio.com/all/__linux__/focal/latest
+ENV TZ=Etc/UTC
+ENV NVBLAS_CONFIG_FILE=/etc/nvblas.conf
+ENV WORKON_HOME=/opt/venv
+ENV PYTHON_VENV_PATH=/opt/venv/reticulate
+ENV PYTHON_CONFIGURE_OPTS=--enable-shared
+ENV RETICULATE_AUTOCONFIGURE=0
+ENV PATH=${PYTHON_VENV_PATH}/bin:${PATH}:${CUDA_HOME}/bin
+
+COPY scripts /rocker_scripts
+
+RUN /rocker_scripts/install_R.sh
+RUN /rocker_scripts/config_R_cuda.sh
+RUN /rocker_scripts/install_python.sh
+
+
+
+
+
diff --git a/stacks/ml-cuda11.1-4.1.0.json b/stacks/ml-cuda11.1-4.1.0.json
@@ -0,0 +1,63 @@
+{
+  "ordered": true,
+  "latest": true,
+  "TAG": "4.1.0-cuda11.1",
+  "LABEL": "org.opencontainers.image.licenses=\"GPL-2.0-or-later\" \\\n      org.opencontainers.image.source=\"https://github.com/rocker-org/rocker-versioned2\" \\\n      org.opencontainers.image.vendor=\"Rocker Project\" \\\n      org.opencontainers.image.authors=\"Carl Boettiger <[email protected]>\"",
+  "stack": [
+  {
+    "IMAGE": "r-ver",
+    "FROM": "nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04",
+    "COPY": "scripts /rocker_scripts",
+    "ENV": {
+      "R_VERSION": "4.1.0",
+      "TERM": "xterm",
+      "LC_ALL": "en_US.UTF-8",
+      "LANG": "en_US.UTF-8",
+      "R_HOME": "/usr/local/lib/R",
+      "CRAN": "https://packagemanager.rstudio.com/all/__linux__/focal/latest",
+      "TZ": "Etc/UTC",
+      "NVBLAS_CONFIG_FILE": "/etc/nvblas.conf",
+      "WORKON_HOME": "/opt/venv",
+      "PYTHON_VENV_PATH": "/opt/venv/reticulate",
+      "PYTHON_CONFIGURE_OPTS": "--enable-shared",
+      "RETICULATE_AUTOCONFIGURE": "0",
+      "PATH": "${PYTHON_VENV_PATH}/bin:${PATH}:${CUDA_HOME}/bin"
+     },
+    "RUN": [
+      "/rocker_scripts/install_R.sh",
+      "/rocker_scripts/config_R_cuda.sh",
+      "/rocker_scripts/install_python.sh"
+    ]
+  },
+  {
+    "IMAGE": "ml",
+    "FROM": "rocker/r-ver:4.1.0-cuda11.1",
+    "ENV": {
+      "S6_VERSION": "v2.1.0.2",
+      "RSTUDIO_VERSION": "preview",
+      "PATH": "/usr/lib/rstudio-server/bin:$PATH"
+
+
+    },
+    "RUN": [
+      "/rocker_scripts/install_rstudio.sh",
+      "/rocker_scripts/install_pandoc.sh",
+      "/rocker_scripts/install_tidyverse.sh"
+      ],
+    "CMD": "[\"/init\"]",
+    "EXPOSE": 8787
+  },
+  {
+    "IMAGE": "ml-verse",
+    "FROM": "rocker/ml:4.1.0-cuda11.1",
+    "RUN": [
+      "/rocker_scripts/install_verse.sh",
+      "/rocker_scripts/install_geospatial.sh"
+      ]
+  }
+  ]
+}
+
+
+
+
diff --git a/stacks/ml-cuda11.1-devel.json b/stacks/ml-cuda11.1-devel.json
@@ -0,0 +1,63 @@
+{
+  "ordered": true,
+  "latest": true,
+  "TAG": "devel-cuda11.1",
+  "LABEL": "org.opencontainers.image.licenses=\"GPL-2.0-or-later\" \\\n      org.opencontainers.image.source=\"https://github.com/rocker-org/rocker-versioned2\" \\\n      org.opencontainers.image.vendor=\"Rocker Project\" \\\n      org.opencontainers.image.authors=\"Carl Boettiger <[email protected]>\"",
+  "stack": [
+  {
+    "IMAGE": "r-ver",
+    "FROM": "nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04",
+    "COPY": "scripts /rocker_scripts",
+    "ENV": {
+      "R_VERSION": "devel",
+      "TERM": "xterm",
+      "LC_ALL": "en_US.UTF-8",
+      "LANG": "en_US.UTF-8",
+      "R_HOME": "/usr/local/lib/R",
+      "CRAN": "https://packagemanager.rstudio.com/all/__linux__/focal/latest",
+      "TZ": "Etc/UTC",
+      "NVBLAS_CONFIG_FILE": "/etc/nvblas.conf",
+      "WORKON_HOME": "/opt/venv",
+      "PYTHON_VENV_PATH": "/opt/venv/reticulate",
+      "PYTHON_CONFIGURE_OPTS": "--enable-shared",
+      "RETICULATE_AUTOCONFIGURE": "0",
+      "PATH": "${PYTHON_VENV_PATH}/bin:${PATH}:${CUDA_HOME}/bin"
+     },
+    "RUN": [
+      "/rocker_scripts/install_R.sh",
+      "/rocker_scripts/config_R_cuda.sh",
+      "/rocker_scripts/install_python.sh"
+    ]
+  },
+  {
+    "IMAGE": "ml",
+    "FROM": "rocker/r-ver:devel-cuda11.1",
+    "ENV": {
+      "S6_VERSION": "v2.1.0.2",
+      "RSTUDIO_VERSION": "preview",
+      "PATH": "/usr/lib/rstudio-server/bin:$PATH"
+
+
+    },
+    "RUN": [
+      "/rocker_scripts/install_rstudio.sh",
+      "/rocker_scripts/install_pandoc.sh",
+      "/rocker_scripts/install_tidyverse.sh"
+      ],
+    "CMD": "[\"/init\"]",
+    "EXPOSE": 8787
+  },
+  {
+    "IMAGE": "ml-verse",
+    "FROM": "rocker/ml:devel-cuda11.1",
+    "RUN": [
+      "/rocker_scripts/install_verse.sh",
+      "/rocker_scripts/install_geospatial.sh"
+      ]
+  }
+  ]
+}
+
+
+
+