LutaoJiang commited on
Commit
6221783
Β·
1 Parent(s): bb7e021
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -79,7 +79,12 @@ def check_gpu():
79
  # test nvdiffrast
80
  import nvdiffrast.torch as dr
81
  dr.RasterizeCudaContext(device="cuda:0")
82
- print("nvdiffrast initialized successfully")
 
 
 
 
 
83
 
84
 
85
  # Only check GPU in non-UI debug mode
 
79
  # test nvdiffrast
80
  import nvdiffrast.torch as dr
81
  dr.RasterizeCudaContext(device="cuda:0")
82
+ print("nvdiffrast initialized successfully")
83
+ subprocess.run(
84
+ shlex.split(
85
+ "rm -rf /home/user/.cache/torch_extensions/py310_cuda121/nvdiffrast_plugin/lock"
86
+ )
87
+ )
88
 
89
 
90
  # Only check GPU in non-UI debug mode