diff --git a/pytorch-2.1.2/recipe/build.sh b/pytorch-2.1.2/recipe/build.sh index 9df3aaa..77857fb 100644 --- a/pytorch-2.1.2/recipe/build.sh +++ b/pytorch-2.1.2/recipe/build.sh @@ -18,12 +18,12 @@ set -ex if [[ $ppc_arch == "p10" ]] then - if [[ -z "${GCC_11_HOME}" ]]; + if [[ -z "${GCC_HOME}" ]]; then - echo "Please set GCC_11_HOME to the install path of gcc-toolset-11" + echo "Please set GCC_HOME to the install path of gcc-toolset-12" exit 1 else - export PATH=${GCC_11_HOME}/bin/:$PATH + export PATH=${GCC_HOME}/bin/:$PATH rm ${PREFIX}/lib/libstdc++.so* rm ${BUILD_PREFIX}/lib/libstdc++.so* fi diff --git a/pytorch-2.1.2/recipe/meta.yaml b/pytorch-2.1.2/recipe/meta.yaml index 01b264f..e36f943 100644 --- a/pytorch-2.1.2/recipe/meta.yaml +++ b/pytorch-2.1.2/recipe/meta.yaml @@ -61,7 +61,6 @@ requirements: - typing_extensions {{ typing_extensions }} - click {{ click }} # https://github.com/AnacondaRecipes/click-feedstock/issues/4 - - importlib-metadata # [py<38] - networkx {{ networkx }} - scipy {{ scipy }} - filelock @@ -85,7 +84,7 @@ requirements: - tensorboard {{ tensorboard }} build: - number: 4 + number: 5 {% if build_type == 'cpu' %} string: {{ build_type }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }} {% else %} @@ -94,7 +93,7 @@ build: detect_binary_files_with_prefix: False script_env: - CUDA_HOME #[build_type == 'cuda'] - - GCC_11_HOME #[ppc_arch == 'p10'] + - GCC_HOME #[ppc_arch == 'p10'] test: imports: