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

Fix explanation

parent da153e7e
No related branches found
No related tags found
No related merge requests found
Pipeline #55921 passed
Stage: build-and-push
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
Dockerfile_cuda
+ 1
− 1
View file @ 28071fac
...@@ -36,7 +36,7 @@ RUN mamba install -c conda-forge -c bioconda -c nvidia --yes \ ...@@ -36,7 +36,7 @@ RUN mamba install -c conda-forge -c bioconda -c nvidia --yes \
torchaudio \ torchaudio \
&& pip install --no-cache-dir --extra-index-url="https://pypi.nvidia.com" \ && pip install --no-cache-dir --extra-index-url="https://pypi.nvidia.com" \
# TensorFlow with CUDA # TensorFlow with CUDA
# CUDA version must not be higher than system: https://www.tensorflow.org/install/source#gpu # CUDA version must not be higher than environment: https://www.tensorflow.org/install/source#gpu
'tensorflow<2.18.0' \ 'tensorflow<2.18.0' \
"nvidia-cuda-nvcc-cu12==${CONDA_OVERRIDE_CUDA}.*" \ "nvidia-cuda-nvcc-cu12==${CONDA_OVERRIDE_CUDA}.*" \
# NVIDIA cuQuantum SDK # NVIDIA cuQuantum SDK
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment