Skip to content

Commit a7903dd

Browse files
committed
update conda config
1 parent 4166ac3 commit a7903dd

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

commons/Dockerfile.base.py38-cuda11.2

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -382,6 +382,7 @@ RUN MPICC=/usr/lib64/openmpi/bin/mpicc python3 -m pip install mpi4py && \
382382
WORKDIR /tmp
383383
COPY ./requirements.22.06.txt /tmp/requirements.txt
384384
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 && \
385386
rm -f /tmp/*.whl /tmp/requirements.txt
386387

387388
RUN python3 -m pip install --extra-index-url \

0 commit comments

Comments
 (0)