update

app.py

@@ -10,20 +10,18 @@ import random
 import math
 import hydra
 import numpy as np
-import torch
-import torch.nn as nn
-print(torch.__version__)
-print(torch.version.cuda)
+# import torch
+# import torch.nn as nn
+# print(torch.__version__)
+# print(torch.version.cuda)
 # import torch.backends.cudnn
-import warp as wp
+# from torch.utils.data import DataLoader
 import glob
-from torch.utils.data import DataLoader
 import os
 import subprocess
 import time
 import cv2
 import copy
-import kornia
 import yaml
 import matplotlib.pyplot as plt
 from sklearn.neighbors import NearestNeighbors
@@ -66,6 +64,13 @@ subprocess.check_call([sys.executable, "-m", "pip", "install", "torch==2.4.0", "
 site.main()  # re-processes every *.pth in site-packages
 importlib.invalidate_caches()
 torch = importlib.import_module("torch")
+
+import torch
+import torch.nn as nn
+from torch.utils.data import DataLoader
+import kornia
+import warp as wp
+
 print(torch.__version__)
 print(torch.version.cuda)