Spaces:

LTT
/

PRM

Runtime error

LTT commited on Nov 26, 2024

Commit

9fd12f3

1 Parent(s): a8f7e29

new

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import spaces
 import os
 os.environ["CXX"] = os.popen("which g++").read().strip()
 os.environ["CC"] = os.popen("which gcc").read().strip()
-os.environ["CUDA_LAUNCH_BLOCKING"] = 1
 import imageio
 import numpy as np
 import torch

 import os
 os.environ["CXX"] = os.popen("which g++").read().strip()
 os.environ["CC"] = os.popen("which gcc").read().strip()
+os.environ["CUDA_LAUNCH_BLOCKING"] = "1"
 import imageio
 import numpy as np
 import torch