diff --git a/ct2_main.py b/ct2_main.py
index 6f4ae6b..194c691 100644
--- a/ct2_main.py
+++ b/ct2_main.py
@@ -7,9 +7,10 @@
 from ct2_utils import CheckQuantizationSupport
 
 def set_cuda_paths():
-    nvidia_base_path = Path(sys.executable).parent / 'Lib' / 'site-packages' / 'nvidia'
-    cublas_bin_path = Path(sys.executable).parent / 'Lib' / 'site-packages' / 'nvidia' / 'cublas' / 'bin'
-    cudnn_bin_path = Path(sys.executable).parent / 'Lib' / 'site-packages' / 'nvidia' / 'cudnn' / 'bin'
+    script_dir = Path(__file__).parent.resolve()
+    nvidia_base_path = script_dir / 'Lib' / 'site-packages' / 'nvidia'
+    cublas_bin_path = script_dir / 'Lib' / 'site-packages' / 'nvidia' / 'cublas' / 'bin'
+    cudnn_bin_path = script_dir / 'Lib' / 'site-packages' / 'nvidia' / 'cudnn' / 'bin'
     
     # Set CUDA_PATH and CUDA_PATH_V12_2
     for env_var in ['CUDA_PATH', 'CUDA_PATH_V12_2']:
@@ -17,11 +18,11 @@ def set_cuda_paths():
         new_paths = [str(nvidia_base_path), str(cublas_bin_path), str(cudnn_bin_path)]
         os.environ[env_var] = os.pathsep.join(filter(None, new_paths + [current_path]))
 
-    # # Add nvidia folder, cudnn bin folder, and cublas bin folder to system PATH
-    # current_path = os.environ.get('PATH', '')
-    # new_paths = [str(nvidia_base_path), str(cublas_bin_path), str(cudnn_bin_path)]
-    # new_path = os.pathsep.join(filter(None, new_paths + [current_path]))
-    # os.environ['PATH'] = new_path
+    # Add nvidia folder, cudnn bin folder, and cublas bin folder to system PATH
+    current_path = os.environ.get('PATH', '')
+    new_paths = [str(nvidia_base_path), str(cublas_bin_path), str(cudnn_bin_path)]
+    new_path = os.pathsep.join(filter(None, new_paths + [current_path]))
+    os.environ['PATH'] = new_path
 
 if __name__ == "__main__":
     set_cuda_paths()