-
Notifications
You must be signed in to change notification settings - Fork 3
Open
Description
The images fail to build in the CI with this error:
docker.io/nvidia/cuda:11.2.1-cudnn8-runtime-ubuntu18.04: not found
Because this image doesn't exist anymore: nvidia/cuda:11.2.1-cudnn8-runtime-ubuntu18.04
(see Dockerhub list here).
Should we go with nvidia/cuda:11.2.2-cudnn8-runtime
@alessandrofasse?
Here's the table of compatibilities for tensorflow and tensorflow-gpu: https://www.tensorflow.org/install/source#gpu
Is there something similar for pytorch?
Metadata
Metadata
Assignees
Labels
No labels