Hi, We were unable to use tf 2.6.0 with the dvc2-base1-gpu docker image. The problem is probably due to the image using cuda version 11.0.3 which is not compatible with tf 2.6.0 (requires 11.2). Are there plans to upgrade the cuda version?

Original question on discord for context.

Many thanks.