try running tests with gpu on ci with cuda 11.2

Author: alonre24

.circleci/config.yml

@@ -388,7 +388,7 @@ jobs:
       enabled: true
       docker_layer_caching: true
       resource_class: gpu.nvidia.small
-      image: ubuntu-1604-cuda-11.2:202012-01
+      image: ubuntu-2004-cuda-11.2:202103-01
 
     steps:
       - build-and-test-gpu-steps