diff --git a/ramalama/common.py b/ramalama/common.py index 1d0c33bd..627fe8d1 100644 --- a/ramalama/common.py +++ b/ramalama/common.py @@ -230,7 +230,7 @@ def get_gpu(): i += 1 if gpu_bytes: - os.environ["HIP_VISIBLE_DEVICES"] = gpu_num + os.environ["HIP_VISIBLE_DEVICES"] = str(gpu_num) return