We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 274a1ec commit f93d2e5Copy full SHA for f93d2e5
builder/manywheel/entrypoint_build.sh
@@ -6,6 +6,8 @@ export PLAT_NAME=$PLAT_NAME
6
export USERID=${USERID}
7
export GROUPID=${GROUPID}
8
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
11
12
source /opt/conda/bin/activate
13
conda activate $PYTHON_VERSION
0 commit comments