diff --git a/vllm/distributed/device_communicators/cuda_wrapper.py b/vllm/distributed/device_communicators/cuda_wrapper.py index bc2cfbf32187..1d53b1c5b809 100644 --- a/vllm/distributed/device_communicators/cuda_wrapper.py +++ b/vllm/distributed/device_communicators/cuda_wrapper.py @@ -5,7 +5,6 @@ """ import ctypes -import glob from dataclasses import dataclass from typing import Any, Dict, List, Optional @@ -62,29 +61,6 @@ def find_loaded_library(lib_name) -> Optional[str]: return path -def get_cudart_lib_path_from_env() -> Optional[str]: - """ - In some system, find_loaded_library() may not work. So we allow users to - specify the path through environment variable VLLM_CUDART_SO_PATH. - """ - cudart_so_env = envs.VLLM_CUDART_SO_PATH - if cudart_so_env is not None: - cudart_paths = [ - cudart_so_env, - ] - for path in cudart_paths: - file_paths = glob.glob(path) - if len(file_paths) > 0: - logger.info( - "Found cudart library at %s through env var" - "VLLM_CUDART_SO_PATH=%s", - file_paths[0], - cudart_so_env, - ) - return file_paths[0] - return None - - class CudaRTLibrary: exported_functions = [ # ​cudaError_t cudaSetDevice ( int device ) @@ -131,7 +107,7 @@ def __init__(self, so_file: Optional[str] = None): if so_file is None: so_file = find_loaded_library("libcudart") if so_file is None: - so_file = get_cudart_lib_path_from_env() + so_file = envs.VLLM_CUDART_SO_PATH # fallback to env var assert so_file is not None, \ ( "libcudart is not loaded in the current process, "