🐇
PoetryでRAPIDSをインストールする
RAPIDSのversionは23.06, cudaのversionは11.0を想定
pyproject.tomlのdepondenciesに以下を記述する
[tool.poetry.dependencies]
cudf-cu11 = { version="^23.06", source="rapids" }
dask-cudf-cu11 = { version="^23.06", source="rapids" }
cuml-cu11 = { version="^23.06", source="rapids" }
ptxcompiler-cu11 = { version="^0.7", source="rapids" }
cubinlinker-cu11 = { version="^0.3", source="rapids" }
rmm-cu11 = { version="^23.06", source="rapids" }
raft-dask-cu11 = { version="^23.06", source="rapids" }
ucx-py-cu11 = { version="^0.32", source="rapids" }
pylibraft-cu11 = { version="^23.06", source="rapids" }
[[tool.poetry.source]]
name="rapids"
url="https://pypi.nvidia.com"
priority="supplemental"
参考
Discussion