Update app.py

app.py

@@ -2,7 +2,9 @@ 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"
+os.environ['CUDA_LAUNCH_BLOCKING']="1"
+os.environ['TORCH_USE_CUDA_DSA'] = "1"
+import nvdiffrast.torch as dr
 import imageio
 import numpy as np
 import torch
@@ -80,15 +82,14 @@ if cuda_path:
     print(f"CUDA installation found at: {cuda_path}")
 else:
     print("CUDA installation not found")
-import os
-
-# 确保 CUDA_HOME 设置正确
-os.environ["CUDA_HOME"] = "/usr/local/cuda"
-os.environ["PATH"] += os.pathsep + os.path.join(os.environ["CUDA_HOME"], "bin")
-os.environ["LD_LIBRARY_PATH"] = os.environ.get("LD_LIBRARY_PATH", "") + os.pathsep + os.path.join(os.environ["CUDA_HOME"], "lib64")
+# import os
 
+# # 确保 CUDA_HOME 设置正确
+# os.environ["CUDA_HOME"] = "/usr/local/cuda"
+# os.environ["PATH"] += os.pathsep + os.path.join(os.environ["CUDA_HOME"], "bin")
+# os.environ["LD_LIBRARY_PATH"] = os.environ.get("LD_LIBRARY_PATH", "") + os.pathsep + os.path.join(os.environ["CUDA_HOME"], "lib64")
+GLCTX = dr.RasterizeCudaContext()
 print(f"GPU: {torch.cuda.is_available()}")
-a = torch.tensor([0]).cuda()
 if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
     device0 = torch.device('cuda:0')
     device1 = torch.device('cuda:0')