diff --git a/ci/test/notebooks.sh b/ci/test/notebooks.sh index a8eccf0a..da8cada9 100644 --- a/ci/test/notebooks.sh +++ b/ci/test/notebooks.sh @@ -6,9 +6,23 @@ set -o pipefail export LIBCUDF_KERNEL_CACHE_PATH=${WORKSPACE}/.jitcache source /opt/conda/bin/activate rapids + +# PyTorch is intentionally excluded from our Docker images due +# to its size, but some notebooks still depend on it. +case "${CUDA_VER}" in +"10.1" | "10.2") + conda install -y -c pytorch pytorch + ;; +*) + echo "Unsupported CUDA version for pytorch." + echo "Not installing pytorch." + ;; +esac + + env /test.sh 2>&1 | tee nbtest.log EXITCODE=$? python /rapids/utils/nbtestlog2junitxml.py nbtest.log -exit ${EXITCODE} \ No newline at end of file +exit ${EXITCODE} diff --git a/conda/recipes/rapids-notebook-env/meta.yaml b/conda/recipes/rapids-notebook-env/meta.yaml index 1988c2c1..279f8be3 100644 --- a/conda/recipes/rapids-notebook-env/meta.yaml +++ b/conda/recipes/rapids-notebook-env/meta.yaml @@ -42,6 +42,7 @@ requirements: - cython {{ cython_version }} - dask-labextension - dask-ml + - filterpy - holoviews - ipython {{ ipython_version }} - jupyter-server-proxy