Correct condition checking cuda dependencies windows (#2629)

Co-authored-by: Hien To <tominhhien97@gmail.com>
This commit is contained in:
hiento09 2024-04-05 16:19:10 +07:00 committed by GitHub
parent 9303e7aeb4
commit b3c8bab153
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -289,7 +289,7 @@ const updateCudaExistence = (
if (process.platform === 'win32') { if (process.platform === 'win32') {
filesCuda12 = ['cublas64_12.dll', 'cudart64_12.dll', 'cublasLt64_12.dll'] filesCuda12 = ['cublas64_12.dll', 'cudart64_12.dll', 'cublasLt64_12.dll']
filesCuda11 = ['cublas64_11.dll', 'cudart64_11.dll', 'cublasLt64_11.dll'] filesCuda11 = ['cublas64_11.dll', 'cudart64_110.dll', 'cublasLt64_11.dll']
paths = process.env.PATH ? process.env.PATH.split(path.delimiter) : [] paths = process.env.PATH ? process.env.PATH.split(path.delimiter) : []
} else { } else {
filesCuda12 = ['libcudart.so.12', 'libcublas.so.12', 'libcublasLt.so.12'] filesCuda12 = ['libcudart.so.12', 'libcublas.so.12', 'libcublasLt.so.12']