在 Google Colab 上安装 NVIDIA Rapids 时出现问题

Question

我正在尝试使用 NVIDIA 提供的以下内容在 Google Colab 的 Tesla P-100CE 上安装 NVIDIA Rapids。

# Installing RAPIDS API.
# NOTE : This may take some few minutes to install.

!git clone https://github.com/rapidsai/rapidsai-csp-utils.git
!bash rapidsai-csp-utils/colab/rapids-colab.sh stable

import sys, os

dist_package_index = sys.path.index('/usr/local/lib/python3.6/dist-packages')
sys.path = sys.path[:dist_package_index] + ['/usr/local/lib/python3.6/site-packages'] + sys.path[dist_package_index:]
sys.path
exec(open('rapidsai-csp-utils/colab/update_modules.py').read(), globals())

执行后，我收到如下错误：

************************************************
Your Google Colab instance has RAPIDS installed!
************************************************
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-2-bdfa5194a5f0> in <module>()
      5 import sys, os
      6 
----> 7 dist_package_index = sys.path.index('/usr/local/lib/python3.6/dist-packages')
      8 sys.path = sys.path[:dist_package_index] + ['/usr/local/lib/python3.6/site-packages'] + sys.path[dist_package_index:]
      9 sys.path

ValueError: '/usr/local/lib/python3.6/dist-packages' is not in list

有人在 Google Colab 中遇到过这个错误吗？如何解决这个问题？感谢这里的任何帮助。

谢谢

Answer 1

2024 年 5 月 14 日更新：

cuDF 和 cudf.pandas 现在预装在 Google Colab 的 GPU 实例上。您仍然可以使用 RAPIDS+Colab Pip 安装模板

将其余 RAPIDS 库（例如 cuML、cuGraph、cuXfilter 和 cuSpatial）pip 安装到启用 GPU 的 Colab 实例上

上一篇：

我们最近推送了一个更新，允许在 Colab 上安装最新的 RAPIDS。您现在可以正常运行 RAPIDS 0.18 和 0.19。可能需要对安装脚本单元进行一些更改，但这些更改记录在我们的文本打印输出中。 尝试一下！ :)