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

[CI] Fix cuDF install; merge 'gpu' and 'cudf' test suite (#5814)

Browse Source
This commit is contained in:
Philip Hyunsu Cho
2020-06-19 01:42:57 -07:00
committed by GitHub
parent a67bc64819
commit a6d9a06b7b
4 changed files with 5 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/ci_build/Dockerfile.gpu
View File

@@ -3,6 +3,7 @@ FROM nvidia/cuda:$CUDA_VERSION-runtime-ubuntu16.04
# Environment
ENV DEBIAN_FRONTEND noninteractive
SHELL ["/bin/bash", "-c"] # Use Bash as shell
# Install all basic requirements
RUN \
@@ -14,11 +15,11 @@ RUN \
ENV PATH=/opt/python/bin:$PATH
# Install Python packages
# Create new Conda environment with cuDF, Dask, and cuPy
RUN \
conda create -n gpu_test -c rapidsai -c nvidia -c conda-forge -c defaults \
python=3.7 dask dask-cuda numpy pytest scipy scikit-learn pandas \
matplotlib wheel python-kubernetes urllib3 graphviz hypothesis
python=3.7 cudf=0.14 cudatoolkit=$CUDA_VERSION dask dask-cuda dask-cudf cupy \
numpy pytest scipy scikit-learn pandas matplotlib wheel python-kubernetes urllib3 graphviz hypothesis
ENV GOSU_VERSION 1.10