From 6a8b0fe9926e8845765a47303a51c906636d9cdf Mon Sep 17 00:00:00 2001 From: Your Name Date: Sat, 22 Jun 2024 03:00:29 +0000 Subject: [PATCH] update pytorch version in default --- finetune/Dockerfile | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/finetune/Dockerfile b/finetune/Dockerfile index 51b4542..eb75db6 100644 --- a/finetune/Dockerfile +++ b/finetune/Dockerfile @@ -80,7 +80,7 @@ ARG CONDA_ENV_NAME="deepspeed" ENV CONDA_ENV_NAME=${CONDA_ENV_NAME} ARG PYTHON_VERSION=3.10 ENV PYTHON_VERSION=${PYTHON_VERSION} -ENV PATH /opt/conda/bin:/opt/conda/envs/${CONDA_ENV_NAME}/bin:$PATH +ENV PATH /opt/conda/bin:/opt/conda/envs/${CONDA_ENV_NAME}/bin:/usr/bin:$PATH ENV REF='main' ENV STAGE_DIR=/tmp ENV NV_PEER_MEM_VERSION=1.2 @@ -89,11 +89,11 @@ ENV OPENMPI_BASEVERSION=4.1 ENV OPENMPI_VERSION=${OPENMPI_BASEVERSION}.6 ARG CUDA='cu121' ENV CUDA=${CUDA} -ARG PYTORCH_VERSION=2.3.0 +ARG PYTORCH_VERSION=2.3.1 ENV PYTORCH_VERSION=${PYTORCH_VERSION} -ARG TORCHVISION_VERSION=0.18.0 +ARG TORCHVISION_VERSION=0.18.1 ENV TORCHVISION_VERSION=${TORCHVISION_VERSION} -ARG TORCHAUDIO_VERSION=2.3.0 +ARG TORCHAUDIO_VERSION=2.3.1 ENV TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION} ARG PYTORCH_CUDA_VERSION=12.1 ENV PYTORCH_CUDA_VERSION=${PYTORCH_CUDA_VERSION}