Merge pull request #1 from ohsu-comp-bio/fix/base-restricted-gpu

provide recent gpu images with readme
ohsu-comp-bio · Apr 12, 2024 · 9db043a · 9db043a
2 parents 0f71efe + aca7fce
commit 9db043a
Show file tree

Hide file tree

Showing 11 changed files with 56 additions and 0 deletions.
diff --git a/.env → deprecated-restricted-gpu/.env b/.env → deprecated-restricted-gpu/.env
diff --git a/Dockerfile → deprecated-restricted-gpu/Dockerfile b/Dockerfile → deprecated-restricted-gpu/Dockerfile
diff --git a/LICENSE → deprecated-restricted-gpu/LICENSE b/LICENSE → deprecated-restricted-gpu/LICENSE
diff --git a/README.md → deprecated-restricted-gpu/README.md b/README.md → deprecated-restricted-gpu/README.md
diff --git a/start-notebook.sh → deprecated-restricted-gpu/start-notebook.sh b/start-notebook.sh → deprecated-restricted-gpu/start-notebook.sh
diff --git a/start-singleuser.sh → ...ecated-restricted-gpu/start-singleuser.sh b/start-singleuser.sh → ...ecated-restricted-gpu/start-singleuser.sh
diff --git a/start.sh → deprecated-restricted-gpu/start.sh b/start.sh → deprecated-restricted-gpu/start.sh
diff --git a/restricted-gpu-slim/Dockerfile b/restricted-gpu-slim/Dockerfile
@@ -0,0 +1,22 @@
+# use base image
+FROM quay.io/ohsu-comp-bio/bridge2ai-jupyter:cpu-root
+
+RUN sudo apt update
+
+# setup CUDA
+RUN sudo apt-get install -y kmod \
+    && wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run \
+    && sudo bash cuda_11.8.0_520.61.05_linux.run --silent --toolkit
+
+# setup cuDNN
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
+    && sudo dpkg -i cuda-keyring_1.1-1_all.deb \
+    && sudo apt-get update \
+    && sudo apt-get -y install cudnn \
+    && echo 'export PATH=$PATH:/usr/local/cuda-11.8/bin' > ~/.bashrc \
+    && echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib6:$LD_LIBRARY_PATH' >> ~/.bashrc \
+    && source ~/.bashrc
+
+# return to non-root user
+USER $NB_UID
+WORKDIR /home/$NB_USER
diff --git a/restricted-gpu-slim/README.md b/restricted-gpu-slim/README.md
@@ -0,0 +1,3 @@
+# Restricted GPU
+
+Restricted GPU with CUDA and CuDNN. Tested on AWS GPU P4 instance (Tesla V100 GPUs). See Dockerfile for more details.
diff --git a/restricted-gpu/Dockerfile b/restricted-gpu/Dockerfile
@@ -0,0 +1,28 @@
+# Use base image
+FROM quay.io/ohsu-comp-bio/bridge2ai-jupyter:cpu-root
+
+RUN sudo apt update
+
+# Setup CUDA
+RUN sudo apt-get install -y kmod \
+    && wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run \
+    && sudo bash cuda_11.8.0_520.61.05_linux.run --silent --toolkit
+
+# Setup cuDNN
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
+    && sudo dpkg -i cuda-keyring_1.1-1_all.deb \
+    && sudo apt-get update \
+    && sudo apt-get -y install cudnn \
+    && echo 'export PATH=$PATH:/usr/local/cuda-11.8/bin' > ~/.bashrc \
+    && echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib6:$LD_LIBRARY_PATH' >> ~/.bashrc \
+    && source ~/.bashrc
+
+# Install TensorFlow
+RUN pip install tensorflow==2.14.0
+
+# Install PyTorch
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+
+# return to non-root user
+USER $NB_UID
+WORKDIR /home/$NB_USER
diff --git a/restricted-gpu/README.md b/restricted-gpu/README.md
@@ -0,0 +1,3 @@
+# Restricted GPU
+
+Restricted GPU with CUDA and CuDNN along with Tensorflow and Pytorch installed. Tested on AWS P3 GPU instance (Tesla V100 GPUs). See Dockerfile for more details.
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		# Restricted GPU

		Restricted GPU with CUDA and CuDNN. Tested on AWS GPU P4 instance (Tesla V100 GPUs). See Dockerfile for more details.
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		# Restricted GPU

		Restricted GPU with CUDA and CuDNN along with Tensorflow and Pytorch installed. Tested on AWS P3 GPU instance (Tesla V100 GPUs). See Dockerfile for more details.