我想知道我是否可以安装 RAPIDS Google Colaboratory notebook 中的库(完全在 GPU 上执行机器学习任务)？

我已经做了一些研究，但我一直无法找到这样做的方法......

最佳答案

现在可以使用新的 T4 实例 https://medium.com/rapids-ai/run-rapids-on-google-colab-for-free-1617ac6323a8

要启用 cuGraph，您可以将 wget 命令替换为:

!conda install -c nvidia/label/cuda10.0 -c rapidsai/label/cuda10.0 -c pytorch\ -c numba -c conda-forge -c numba -c 默认值\ 提升 cudf=0.6 cuml=0.6 python=3.6 cugraph=0.6 -y

关于google-colaboratory - 在 Google Colab notebook 上安装 RAPIDS 库，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/53813125/