[Draft] AMD ROCm support

[benoit-cty]

Description

Continuing #490

Related Issue

Please link to the issue this PR resolves: [issue #178 ]

Motivation and Context

AMD GPU are not yet supported.

How Has This Been Tested?

Using Adastra supercomputer. With AMD MI250 GPUs.

Types of changes

What types of changes does your code introduce? Put an x in all the boxes that apply:

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)

Checklist:

Go over all the following points, and put an x in all the boxes that apply.

• My code follows the code style of this project.
• My change requires a change to the documentation.
• I have updated the documentation accordingly.
• I have read the CONTRIBUTING.md document.
• I have added tests to cover my changes.
• All new and existing tests passed.

[benoit-cty]

I manage to make CodeCarbon works on Adastra and upgrade the @IlyasMoutawwakil code to support more recent version of the amdsmi package.

There is still work to do as the metrics are weird:

[codecarbon INFO @ 17:04:13] Energy consumed for all GPUs : 4.254300 kWh. Total GPU Power : 12572969.923258875 W