Dockerfile for modern GPUS

[kurtjcu]

Describe the feature
Would be great to be able to run this on modern hardware. Current dockerfile targets 6.0, 6.1, 7.0 which is pre 20 series/ Turing(released 2018). To work with modern systems it would make sense to be compatible from 7.0 to 9.0/9.5.

Motivation
Current hardware target may not be available to purchase due to age and modern hardware is faster.

Related resources
Will have to update base pytorch container to have a CUDA version compatible with modern hardware as well.

Additional context
May find issues compiling on current versions of pytorch/cuda due to age of some of the libraries used.