Update whispers2t_batch_gui.py

Make setting cuda-related paths more robust.

whispers2t_batch_gui.py

@@ -26,16 +26,29 @@
 from whispers2t_batch_transcriber import Worker
 
 def set_cuda_paths():
-    try:
-        venv_base = Path(sys.executable).parent
-        nvidia_base_path = venv_base / 'Lib' / 'site-packages' / 'nvidia'
-        for env_var in ['CUDA_PATH', 'CUDA_PATH_V12_1', 'PATH']:
-            current_path = os.environ.get(env_var, '')
-            os.environ[env_var] = os.pathsep.join(filter(None, [str(nvidia_base_path), current_path]))
-        logging.info("CUDA paths set successfully")
-    except Exception as e:
-        logging.error(f"Error setting CUDA paths: {str(e)}")
-        logging.debug(traceback.format_exc())
+    venv_base = Path(sys.executable).parent.parent
+    nvidia_base_path = venv_base / 'Lib' / 'site-packages' / 'nvidia'
+    cuda_path = nvidia_base_path / 'cuda_runtime' / 'bin'
+    cublas_path = nvidia_base_path / 'cublas' / 'bin'
+    cudnn_path = nvidia_base_path / 'cudnn' / 'bin'
+    paths_to_add = [str(cuda_path), str(cublas_path), str(cudnn_path)]
+    env_vars = ['CUDA_PATH', 'CUDA_PATH_V12_1', 'PATH']
+
+    # print(f"Virtual environment base: {venv_base}")
+    # print(f"NVIDIA base path: {nvidia_base_path}")
+    # print(f"CUDA path: {cuda_path}")
+    # print(f"cuBLAS path: {cublas_path}")
+    # print(f"cuDNN path: {cudnn_path}")
+
+    for env_var in env_vars:
+        current_value = os.environ.get(env_var, '')
+        new_value = os.pathsep.join(paths_to_add + [current_value] if current_value else paths_to_add)
+        os.environ[env_var] = new_value
+        # print(f"\n{env_var} updated:")
+        # print(f"  Old value: {current_value}")
+        # print(f"  New value: {new_value}")
+
+    # print("\nCUDA paths have been set or updated in the environment variables.")
 
 set_cuda_paths()
 
@@ -221,4 +234,4 @@ def workerFinished(self, message):
     app.setStyle("Fusion")
     mainWindow = MainWindow()
     mainWindow.show()
-    sys.exit(app.exec())
+    sys.exit(app.exec())