Spaces:

LTT
/

PRM

Running on Zero

LTT commited on Nov 26, 2024

Commit

a8f7e29

•

1 Parent(s): b891190

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()
 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