Rapids

来自Shiyin's note
跳到导航 跳到搜索

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