This commit is contained in:
parent
51b5360ee4
commit
1fe51f208d
|
|
@ -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
|
python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy pyyaml
|
||||||
|
|
||||||
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