How to Select/Isolate GPUs when Using ROCm in Python

Question

Is their an equivalent for os.environ["CUDA_VISIBLE_DEVICES"]="0" in the os module for isolating GPUs when using AMD ROCm?

Answer 1

According to this documentation, HIP_VISIBLE_DEVICES environment variable should have the same effect. So:

os.environ["HIP_VISIBLE_DEVICES"]="0"