Skip to content

Commit f93d2e5

Browse files
authored
fix turbomind lib missing to link nccl by exporting nccl path (#3479)
1 parent 274a1ec commit f93d2e5

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

builder/manywheel/entrypoint_build.sh

+2
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@ export PLAT_NAME=$PLAT_NAME
66
export USERID=${USERID}
77
export GROUPID=${GROUPID}
88
export CUDAVER=$(nvcc --version | sed -n 's/^.*release \([0-9]\+\).*$/\1/p')
9+
export NCCL_INCLUDE_DIR=/usr/local/cuda/include
10+
export NCCL_LIB_DIR=/usr/local/cuda/lib64
911

1012
source /opt/conda/bin/activate
1113
conda activate $PYTHON_VERSION

0 commit comments

Comments
 (0)