Upload script.py

script.py

@@ -207,21 +207,33 @@ def setup_environment():
 	# download_packages(['scikit-learn'], folder='packages/scikit-learn')
 	# download_packages(['open3d'], folder='packages/open3d')
 	# download_packages(['easydict'], folder='packages/easydict')
+	
+	setup_cuda_environment()
 
 	pc_util_path = os.path.join(os.getcwd(), 'pc_util')
 	if os.path.isdir(pc_util_path):
 		os.chdir(pc_util_path)
 		subprocess.check_call([sys.executable, "setup.py", "install"])
+		os.chdir("..")
 	else:
 		print(f"Directory {pc_util_path} does not exist")
 	
 def setup_cuda_environment():
-	cuda_home = '/usr/local/cuda-12.1'
+	cuda_home = '/usr/local/cuda'
 	if not os.path.exists(cuda_home):
 		raise EnvironmentError(f"CUDA_HOME directory {cuda_home} does not exist. Please install CUDA and set CUDA_HOME environment variable.")
-	os.environ['CUDA_HOME'] = cuda_home
-	os.environ['PATH'] = f"{cuda_home}/bin:{os.environ['PATH']}"
-	os.environ['LD_LIBRARY_PATH'] = f"{cuda_home}/lib64:{os.environ.get('LD_LIBRARY_PATH', '')}"
+	# os.environ['CUDA_HOME'] = cuda_home
+	# os.environ['PATH'] = f"{cuda_home}/bin:{os.environ['PATH']}"
+	# os.environ['LD_LIBRARY_PATH'] = f"{cuda_home}/lib64:{os.environ.get('LD_LIBRARY_PATH', '')}"
+
+	os.environ['PATH'] = '/usr/local/cuda/bin'
+	os.environ['LD_LIBRARY_PATH'] = '/usr/local/cuda/lib64'  
+	os.environ['LIBRARY_PATH'] = '/usr/local/cuda/lib64'
+
+	usr_local_contents = os.listdir('/usr/local')
+	# print("Items under /usr/local:")
+	for item in usr_local_contents:
+		print(item)
 
 from pathlib import Path
 def save_submission(submission, path):