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

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
4 changed files with 5 additions and 4 deletions

View File

@@ -29,7 +29,7 @@ ENV PATH=/opt/python/bin:$PATH
# Create new Conda environment with RMM
RUN \
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