Disable AMD memory benchmarks (#29871)

* remove py3nvml to skip amd memory benchmarks

* uninstall pynvml from docker images

docker/transformers-pytorch-amd-gpu/Dockerfile

@@ -34,3 +34,6 @@ RUN python3 -m pip uninstall -y tensorflow flax
 # When installing in editable mode, `transformers` is not recognized as a package.
 # this line must be added in order for python to be aware of transformers.
 RUN cd transformers && python3 setup.py develop
+
+# Remove nvml as it is not compatible with ROCm
+RUN python3 -m pip uninstall py3nvml pynvml -y

docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile

@@ -43,3 +43,6 @@ RUN python3 -m pip install --no-cache-dir ./transformers[accelerate,testing,sent
 RUN cd transformers && python3 setup.py develop
 
 RUN python3 -c "from deepspeed.launcher.runner import main"
+
+# Remove nvml as it is not compatible with ROCm
+RUN python3 -m pip uninstall py3nvml pynvml -y