This commit is contained in:
parent
1fe51f208d
commit
e337a6fdd3
|
|
@ -15,7 +15,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
libnccl2=2.22.3-1+cuda12.6 \
|
libnccl2=2.22.3-1+cuda12.6 \
|
||||||
libnccl-dev=2.22.3-1+cuda12.6 \
|
libnccl-dev=2.22.3-1+cuda12.6 \
|
||||||
libjpeg-dev libpng-dev ca-certificates && \
|
libjpeg-dev libpng-dev ca-certificates && \
|
||||||
python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy pyyaml
|
python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy pyyaml typing-extensions
|
||||||
|
|
||||||
WORKDIR /opt
|
WORKDIR /opt
|
||||||
RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git
|
RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue