From da6d552f5641a4a8eef7a6abf1ad3cad61f9e8e3 Mon Sep 17 00:00:00 2001 From: Sarah Yurick Date: Thu, 19 Sep 2024 11:38:20 -0700 Subject: [PATCH] python3 dev Signed-off-by: Sarah Yurick --- .github/workflows/gpuci.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/gpuci.yml b/.github/workflows/gpuci.yml index 18b05423..0210e31e 100644 --- a/.github/workflows/gpuci.yml +++ b/.github/workflows/gpuci.yml @@ -28,7 +28,8 @@ jobs: - name: Install NeMo-Curator and PyTest run: | - pip install wheel cython fasttext-wheel + sudo apt-get install -y python3-dev + pip install wheel cython RAPIDS_NIGHTLY=1 pip install --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple ".[cuda12x]" pip install pytest