We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4166ac3 commit a7903ddCopy full SHA for a7903dd
commons/Dockerfile.base.py38-cuda11.2
@@ -382,6 +382,7 @@ RUN MPICC=/usr/lib64/openmpi/bin/mpicc python3 -m pip install mpi4py && \
382
WORKDIR /tmp
383
COPY ./requirements.22.06.txt /tmp/requirements.txt
384
RUN python3 -m pip install --no-cache-dir -r requirements.txt && \
385
+ python3 -m pip install --no-cache-dir torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu112 && \
386
rm -f /tmp/*.whl /tmp/requirements.txt
387
388
RUN python3 -m pip install --extra-index-url \
0 commit comments