chore(deps): update nvidia/cuda docker tag to v11.3.1 (GoogleCloudPlatform#6089)

renovate-bot · web-flow · commit e5233145153a · 2021-06-06T16:31:13.000-04:00
diff --git a/dataflow/gpu-workers/Dockerfile b/dataflow/gpu-workers/Dockerfile
@@ -16,7 +16,7 @@
 #   https://www.tensorflow.org/install/source#gpu
 # Check the Nvidia container registry catalog to look at the available Nvidia images:
 #   https://ngc.nvidia.com/catalog/containers/nvidia:cuda
-FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04
+FROM nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04
 
 # The Python version of the Dockerfile MUST match the Python version you use
 # to launch the Dataflow job.
