“Rapids”的版本间差异

来自Shiyin's note
跳到导航 跳到搜索
(创建页面,内容为“Nvidia出的python下的GPU加速套件 [https://rapids.ai/start.html] *cudf *cuML 加速scikit-learn [https://docs.rapids.ai/api/cuml/stable/] *安装(兼顾pyt…”)
 
(没有差异)

2022年2月8日 (二) 08:03的最新版本

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