We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7abd789 commit bdbeef5Copy full SHA for bdbeef5
2 files changed
Dockerfile
@@ -55,8 +55,7 @@ RUN wget -O /tmp/miniconda3.sh \
55
conda activate base && \
56
conda upgrade libstdcxx-ng -c conda-forge -y && \
57
conda install -y nvidia/label/cuda-${CUDA_VERSION}::cuda-toolkit && \
58
- echo "y" | conda install -y nvidia/label/cudnn-${CUDNN_VERSION}::cudnn && \
59
- bash /tmp/conda_essentials.sh && \
+ conda install -y nvidia/label/cudnn-${CUDNN_VERSION}::cudnn && \
60
pip3 install torch==${TORCH_VERSION} torchvision torchaudio \
61
--index-url https://download.pytorch.org/whl/cu126 \
62
--no-cache-dir
scripts/.image-configs.sh
@@ -1,4 +1,4 @@
1
-IMAGE_VERSION=2.0.2
+IMAGE_VERSION=2.0.3
2
TORCH_VERSION=2.7.0
3
CUDA_VERSION=12.6.0
4
CUDNN_VERSION=9.8.0
0 commit comments