Is your feature request related to a problem? Please describe.
No

Describe the solution you'd like
Support for running AMD GPUs in the container.

Describe alternatives you've considered
N/A

Additional context
Need AMD GPU for testing.