在Googe Colab笔记本上安装RAPIDS库

Nov*_*vak 1 google-colaboratory rapids

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

我已经做过一些研究,但是我找不到解决方法...

Jos*_*shP 8

现在,可以使用新的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默认值\ boost cudf = 0.6 cuml = 0.6 python = 3.6 cugraph = 0.6 -y