code crashes for rocm added proper type cast for env var

Signed-off-by: Brian <[email protected]>
containers · Jan 20, 2025 · 964a8ea · 964a8ea
1 parent 25168aa
commit 964a8ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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