File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff line change @@ -422,12 +422,11 @@ RUN python3 -m pip --no-cache-dir install git+https://github.com/rpy2/rpy2.git@$
422
422
rm -f /tmp/*.whl /tmp/requirements.txt
423
423
424
424
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 && \
428
427
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
431
430
432
431
RUN CUDA_ROOT=/usr/local/cuda python3 -m pip install --no-cache-dir \
433
432
mpi4jax \
You can’t perform that action at this time.
0 commit comments