diff --git a/recipe/build.sh b/recipe/build.sh index 477ba3f2..d8655ef4 100644 --- a/recipe/build.sh +++ b/recipe/build.sh @@ -147,7 +147,9 @@ elif [[ ${cuda_compiler_version} != "None" ]]; then # with no NVIDIA GPUs. export USE_MKLDNN=1 export USE_CUDA=1 - export USE_CUFILE=1 + if [[ "$target_platform" != "linux-ppc64le" ]]; then + export USE_CUFILE=1 + fi # PyTorch has multiple different bits of logic finding CUDA, override # all of them. export CUDAToolkit_BIN_DIR=${BUILD_PREFIX}/bin diff --git a/recipe/meta.yaml b/recipe/meta.yaml index 77e0a4c8..bc8a1bb3 100644 --- a/recipe/meta.yaml +++ b/recipe/meta.yaml @@ -101,7 +101,7 @@ requirements: - cuda-nvml-dev - cuda-profiler-api - libcublas-dev - - libcufile-dev + - libcufile-dev # [not ppc64le] - libcufft-dev - libcurand-dev - libcusolver-dev @@ -218,7 +218,7 @@ outputs: - cuda-nvml-dev - cuda-profiler-api - libcublas-dev - - libcufile-dev + - libcufile-dev # [not ppc64le] - libcufft-dev - libcurand-dev - libcusolver-dev