NCCL dirs have to be specified. Otherwise picks up different version

Author: syed-ahmed

manywheel/build_cuda.sh

@@ -209,6 +209,8 @@ elif [[ $CUDA_VERSION == "11.7" ]]; then
         export ATEN_STATIC_CUDA=0
         export USE_CUDA_STATIC_LINK=0
         export USE_CUPTI_SO=1
+        export NCCL_INCLUDE_DIR="/usr/local/cuda/include/"
+        export NCCL_LIB_DIR="/usr/local/cuda/lib64/"
     fi
 else
     echo "Unknown cuda version $CUDA_VERSION"