Skip to content

Instantly share code, notes, and snippets.

@eugeneyan

eugeneyan/nvt.yml

Created December 15, 2021 00:27
Show Gist options
  • Save eugeneyan/3435e05dd675b9ee2af164214536752d to your computer and use it in GitHub Desktop.
Save eugeneyan/3435e05dd675b9ee2af164214536752d to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. eugeneyan created this gist Dec 15, 2021.
    22 changes: 22 additions & 0 deletions nvt.yml
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,22 @@
    # Based on https://github.com/NVIDIA-Merlin/NVTabular/blob/main/conda/environments/nvtabular_dev_cuda11.0.yml
    name: nvt_t4r
    channels:
    - rapidsai
    - nvidia
    - conda-forge
    - defaults
    dependencies:
    - nvtabular
    - python>=3.7
    - cudatoolkit=11.0
    - cudf>=21.08.*
    - dask-cuda>=21.08.*
    - dask-cudf>=21.08.*
    - rmm>=21.08.*
    - dask==2021.7.1
    - distributed>=2021.7.1
    - nvtx>=0.2.1
    - numba>=0.53.0
    - dlpack
    - scikit-learn
    - asvdb