Updating wheels with cuda 11

[scarrazza]

Here we update the wheels build procedure to include cuda 11. For simplicity I am launching the wheels action for all branches.

[codecov]

Codecov Report

Merging #301 (c78f7f8) into master (4b2c72b) will not change coverage.
The diff coverage is n/a.

@@            Coverage Diff            @@
##            master      #301   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files           56        56           
  Lines        10716     10716           
=========================================
  Hits         10716     10716           

Flag	Coverage Δ
unittests	100.00% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 4b2c72b...c78f7f8. Read the comment docs.

[stavros11]

Thanks for the update. Note that I still don't have CUDA 11 on my local machine so I could not test the wheel on GPU, but it works on CPU. By the way I tried installing the python38 wheel on dom that has CUDA 11 but installation fails probably because it has Python 3.9.

[scarrazza]

Thanks for checking, I have tested the latest wheel package on dom inside a conda environment with tf2.4.0 and cuda 11 and all tests are passing on GPU.