fix which command on linux server (#523)

mindspore-lab · May 22, 2023 · 775cda4 · 775cda4
1 parent 54072ef
commit 775cda4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mindnlp/models/utils/kernel_utils.py b/mindnlp/models/utils/kernel_utils.py
@@ -24,7 +24,7 @@ def _find_cuda_home():
     cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
     if cuda_home is None:
         try:
-            nvcc = subprocess.check_output(['where', 'nvcc']).decode().rstrip('\r\n')
+            nvcc = subprocess.check_output(['which', 'nvcc']).decode().rstrip('\r\n')
             cuda_home = os.path.dirname(os.path.dirname(nvcc))
         except subprocess.CalledProcessError as exc:
             raise RuntimeError('NVCC Not Available') from exc