diff --git a/Dockerfile b/Dockerfile index 15c7889..1cd4961 100644 --- a/Dockerfile +++ b/Dockerfile @@ -15,7 +15,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ libnccl2=2.22.3-1+cuda12.6 \ libnccl-dev=2.22.3-1+cuda12.6 \ libjpeg-dev libpng-dev ca-certificates && \ - python3 -m pip install --no-cache-dir --upgrade pip wheel setuptoo + python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools WORKDIR /opt RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git