diff --git a/text-0.16.2/recipe/meta.yaml b/text-0.16.2/recipe/meta.yaml index db808fc..5a98ef9 100644 --- a/text-0.16.2/recipe/meta.yaml +++ b/text-0.16.2/recipe/meta.yaml @@ -13,7 +13,7 @@ source: - 0101-disable-tcmalloc.patch build: - number: 2 + number: 3 string: {{ build_type }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cpu'] string: {{ build_type }}{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cuda'] script_env: #[ppc_arch == 'p10'] @@ -29,6 +29,7 @@ requirements: - python {{ python }} - _pytorch_select {{ pytorch_select_version }} - pytorch-base {{ pytorch }} + - cudatoolkit {{ cudatoolkit }} #[build_type == 'cuda'] run: - numpy {{ numpy }} - python {{ python }} @@ -38,7 +39,7 @@ requirements: - tqdm {{ tqdm }} - sentencepiece {{ sentencepiece }} - torchdata - + - cudatoolkit {{ cudatoolkit }} #[build_type == 'cuda'] test: imports: - torchtext diff --git a/text-0.16.2/variants/meta.yaml b/text-0.16.2/variants/meta.yaml index 8581d5b..5df8c71 100644 --- a/text-0.16.2/variants/meta.yaml +++ b/text-0.16.2/variants/meta.yaml @@ -7,7 +7,7 @@ package: version: {{ version }} build: - number: 1 + number: 2 string: py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cpu'] string: {{ build_type }}{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cuda'] @@ -17,6 +17,7 @@ requirements: - _pytorch_select {{ pytorch_select_version }} - torchtext-base {{ torchtext }} - python {{ python }} + - cudatoolkit {{ cudatoolkit }} #[build_type == 'cuda'] about: home: http://pytorch.org/