Skip to content

Create conda template for GPU dependencies. #63

@trivialfis

Description

@trivialfis

We can use a jinja template to avoid:

    mamba create -y -n gpu_test -c rapidsai -c conda-forge -c nvidia \
        python=$PYTHON_VERSION "cudf=$RAPIDS_VERSION.*" "rmm=$RAPIDS_VERSION.*" cuda-version=$CUDA_SHORT_VER \
        "nccl=${NCCL_SHORT_VER}" \
        dask \
        distributed \
        "dask-cuda=$RAPIDS_VERSION.*" "dask-cudf=$RAPIDS_VERSION.*" cupy \
        numpy pytest pytest-timeout scipy scikit-learn pandas matplotlib wheel \
        python-kubernetes urllib3 graphviz hypothesis "loky>=3.5.1" \
        "pyspark>=3.4.0,<4.0" cloudpickle cuda-python && \

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions