Fix docker build script to build with CUDA 11.7

[agunapal]

Description

Fixing the build script to build docker image with CUDA 11.7

Fixes #(issue)

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)

Feature/Issue validation/testing

docker image details

Checklist:

• Did you have fun?
• Have you added tests that prove your fix is effective or that this feature works?
• Has code been commented, particularly in hard-to-understand areas?
• Have you made corresponding changes to the documentation?

[msaroufim]

The new flakiness with ubuntu 20.04 is bugging me, curious does it go away if we switch to 22.04?

[codecov]

Codecov Report

Merging #2032 (298acf5) into master (ddaddb9) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##           master    #2032   +/-   ##
=======================================
  Coverage   53.39%   53.39%           
=======================================
  Files          71       71           
  Lines        3223     3223           
  Branches       56       56           
=======================================
  Hits         1721     1721           
  Misses       1502     1502           

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more