[backport] Update CUDA docker image and NCCL. (#8139) (#8162)

* Update CUDA docker image and NCCL. (#8139) * Rest of the CI. * CPU test dependencies.
2022-08-12 18:57:42 +08:00
parent a55d3bdde2
commit 39c1488a42
13 changed files with 68 additions and 45 deletions
--- a/tests/ci_build/conda_env/cpu_test.yml
+++ b/tests/ci_build/conda_env/cpu_test.yml
@@ -30,15 +30,13 @@ dependencies:
 - jsonschema
 - boto3
 - awscli
- numba
- llvmlite
 - py-ubjson
 - cffi
 - pyarrow
- protobuf<=3.20
+- protobuf
+- pyspark>=3.3.0
+- cloudpickle
+- shap
+- modin
 - pip:
-  - shap
-  - ipython                     # required by shap at import time.
-  - sphinx_rtd_theme
  - datatable
-  - modin[all]