Skip to content

Commit 666f4fb

Browse files
committed
nvidia-dali-tf-plugin-cuda110
1 parent 94c1453 commit 666f4fb

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

python-ff/Dockerfile.conda.ff.23.01

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -422,12 +422,11 @@ RUN python3 -m pip --no-cache-dir install git+https://github.com/rpy2/rpy2.git@$
422422
rm -f /tmp/*.whl /tmp/requirements.txt
423423

424424
RUN python3 -m pip install --extra-index-url \
425-
https://developer.download.nvidia.com/compute/redist \
426-
--upgrade nvidia-dali-cuda110 \
427-
&& \
425+
https://developer.download.nvidia.com/compute/redist/ \
426+
nvidia-dali-cuda110==1.22.0 && \
428427
python3 -m pip install --extra-index-url \
429-
https://developer.download.nvidia.com/compute/redist \
430-
--upgrade nvidia-dali-tf-plugin-cuda110
428+
https://developer.download.nvidia.com/compute/redist/ \
429+
nvidia-dali-tf-plugin-cuda110==1.22.0
431430

432431
RUN CUDA_ROOT=/usr/local/cuda python3 -m pip install --no-cache-dir \
433432
mpi4jax \

0 commit comments

Comments
 (0)