Correct condition checking cuda dependencies windows (#2629)
Co-authored-by: Hien To <tominhhien97@gmail.com>
This commit is contained in:
parent
9303e7aeb4
commit
b3c8bab153
@ -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']
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user