|
|
#!/bin/bash |
|
|
python -m pip install git+https://github.com/huggingface/transformers.git |
|
|
python -m pip install pytorch-crf |
|
|
python -m pip install --upgrade pip |
|
|
python -m pip install -r requirements.txt |
|
|
python -m pip install \ |
|
|
--extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple \ |
|
|
"cudf-cu12>=24.4.0a0,<=24.4" "dask-cudf-cu12>=24.4.0a0,<=24.4" \ |
|
|
"cuml-cu12>=24.4.0a0,<=24.4" "cugraph-cu12>=24.4.0a0,<=24.4" \ |
|
|
"cuspatial-cu12>=24.4.0a0,<=24.4" "cuproj-cu12>=24.4.0a0,<=24.4" \ |
|
|
"cuxfilter-cu12>=24.4.0a0,<=24.4" "cucim-cu12>=24.4.0a0,<=24.4" \ |
|
|
"pylibraft-cu12>=24.4.0a0,<=24.4" "raft-dask-cu12>=24.4.0a0,<=24.4" \ |
|
|
"dask-cuda>=24.4.0a0,<=24.4" |
|
|
python -m pip install poetry |
|
|
|
