We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see ourdocumentation.
There was an error while loading.Please reload this page.
1 parent30f2cf9 commit71550acCopy full SHA for 71550ac
pyproject.toml
@@ -28,11 +28,14 @@ before-all = [
28
"chmod +x scripts/build-env/manylinux.sh",
29
"scripts/build-env/manylinux.sh"
30
]
31
+repair-wheel-command = [
32
+"auditwheel repair -w {dest_dir} --exclude libcudart.so.* --exclude libcublas.so.* --exclude libcublasLt.so.* {wheel}"
33
+]
34
35
[tool.cibuildwheel.linux.environment]
36
cuda ="12.4"
37
temp ="/tmp/cuda-temp"
38
TENSOR_ARRAY_INSTALL_PATH ="/tmp/tensor-array-temp"
39
CUDA_PATH ="/usr/local/cuda-${cuda}"
40
PATH ="${PATH}:${CUDA_PATH}/bin"
-LD_LIBRARY_PATH ="${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64"
41
+LD_LIBRARY_PATH ="${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64:${TENSOR_ARRAY_INSTALL_PATH}/lib/tensor-array:${TENSOR_ARRAY_INSTALL_PATH}/lib64/tensor-array"