Rapids

来自Shiyin's note
Shen讨论 | 贡献2022年2月8日 (二) 08:03的版本 (创建页面,内容为“Nvidia出的python下的GPU加速套件 [https://rapids.ai/start.html] *cudf *cuML 加速scikit-learn [https://docs.rapids.ai/api/cuml/stable/] *安装(兼顾pyt…”)
(差异) ←上一版本 | 最后版本 (差异) | 下一版本→ (差异)
跳到导航 跳到搜索

Nvidia出的python下的GPU加速套件 [1]

  • cudf
  • cuML 加速scikit-learn [2]
  • 安装(兼顾pytorch的兼容性)conda create -n pytorch_and_rapids -c rapidsai-nightly -c nvidia -c pytorch -c conda-forge rapids=22.04 pytorch=1.10.0 cudatoolkit=11.3