Merge pull request #602 from containers/rocm-hotfix

code crashes for rocm added proper type cast for env var
containers · Jan 21, 2025 · e08af09 · e08af09
2 parents 25168aa + 964a8ea
commit e08af09
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