Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CI] Configure RAPIDS, dask, modin (#7033)

Browse Source 
Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan 2021-06-18 10:27:51 +08:00 committed by GitHub
parent d9799b09d0
commit dcd84b3979
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/ci_build/Dockerfile.gpu
View File

@ -19,7 +19,7 @@ ENV PATH=/opt/python/bin:$PATH
# Create new Conda environment with cuDF, Dask, and cuPy # Create new Conda environment with cuDF, Dask, and cuPy
RUN \ RUN \
conda create -n gpu_test -c rapidsai-nightly -c rapidsai -c nvidia -c conda-forge -c defaults \ conda create -n gpu_test -c rapidsai-nightly -c rapidsai -c nvidia -c conda-forge -c defaults \
python=3.7 cudf=21.06* rmm=21.06* cudatoolkit=$CUDA_VERSION_ARG dask=2021.05.0 dask-cuda dask-cudf cupy \ python=3.7 cudf=21.08* rmm=21.08* cudatoolkit=$CUDA_VERSION_ARG dask dask-cuda dask-cudf cupy \
numpy pytest scipy scikit-learn pandas matplotlib wheel python-kubernetes urllib3 graphviz hypothesis numpy pytest scipy scikit-learn pandas matplotlib wheel python-kubernetes urllib3 graphviz hypothesis
ENV GOSU_VERSION 1.10 ENV GOSU_VERSION 1.10

2
tests/ci_build/Dockerfile.rmm
View File

@ -29,7 +29,7 @@ ENV PATH=/opt/python/bin:$PATH
# Create new Conda environment with RMM # Create new Conda environment with RMM
RUN \ RUN \
conda create -n gpu_test -c rapidsai-nightly -c rapidsai -c nvidia -c conda-forge -c defaults \ conda create -n gpu_test -c rapidsai-nightly -c rapidsai -c nvidia -c conda-forge -c defaults \
python=3.7 rmm=21.06* cudatoolkit=$CUDA_VERSION_ARG python=3.7 rmm=21.08* cudatoolkit=$CUDA_VERSION_ARG
ENV GOSU_VERSION 1.10 ENV GOSU_VERSION 1.10

4
tests/ci_build/conda_env/macos_cpu_test.yml
View File

@ -13,8 +13,8 @@ dependencies:
- scikit-learn - scikit-learn
- pandas - pandas
- matplotlib - matplotlib
- dask=2021.05.0 - dask
- distributed=2021.05.0 - distributed
- graphviz - graphviz
- python-graphviz - python-graphviz
- hypothesis - hypothesis

1
tests/ci_build/test_python.sh
View File

@ -60,6 +60,7 @@ case "$suite" in
cpu) cpu)
source activate cpu_test source activate cpu_test
install_xgboost install_xgboost
export RAY_OBJECT_STORE_ALLOW_SLOW_STORAGE=1
pytest -v -s -rxXs --fulltrace --durations=0 ${args} tests/python pytest -v -s -rxXs --fulltrace --durations=0 ${args} tests/python
cd tests/distributed cd tests/distributed
./runtests.sh ./runtests.sh