Skip to content
Snippets Groups Projects
Commit 2c8675fc authored by Seungmin Kim's avatar Seungmin Kim
Browse files

Fix tag

parent 754bfaeb
No related branches found
No related tags found
No related merge requests found
Pipeline #55707 passed
ARG BASE_CONTAINER=quay.io/jupyter/tensorflow-notebook:cuda-2025-01-20
ARG BASE_CONTAINER=quay.io/jupyter/tensorflow-notebook:cuda-2024-10-22
FROM $BASE_CONTAINER
LABEL maintainer="Dima Mishin <dmishin@ucsd.edu>, Seungmin Kim <ehf@yonsei.ac.kr>"
......@@ -25,7 +25,7 @@ RUN mamba install -c conda-forge --yes \
# NVIDIA cuQuantum SDK
'cuquantum' \
'cuquantum-python' \
'cuda-version=12' \
'cuda-version=12.4' \
&& mamba clean --all -f -y \
&& pip install --no-cache-dir --extra-index-url="https://pypi.nvidia.com" \
"tensorflow[and-cuda]==$(python3 -c 'import tensorflow as tf; print(tf.__version__)' 2>/dev/null)" \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment