Skip to content

Commit bdbeef5

Browse files
committed
Fix Dockerfile
1 parent 7abd789 commit bdbeef5

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

Dockerfile

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,7 @@ RUN wget -O /tmp/miniconda3.sh \
5555
conda activate base && \
5656
conda upgrade libstdcxx-ng -c conda-forge -y && \
5757
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 && \
58+
conda install -y nvidia/label/cudnn-${CUDNN_VERSION}::cudnn && \
6059
pip3 install torch==${TORCH_VERSION} torchvision torchaudio \
6160
--index-url https://download.pytorch.org/whl/cu126 \
6261
--no-cache-dir

scripts/.image-configs.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
IMAGE_VERSION=2.0.2
1+
IMAGE_VERSION=2.0.3
22
TORCH_VERSION=2.7.0
33
CUDA_VERSION=12.6.0
44
CUDNN_VERSION=9.8.0

0 commit comments

Comments
 (0)