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

Question

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

# 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

我们最近推出了一个更新，允许在Colab上安装最新的RAPIDS。您现在可以像往常一样运行RAPIDS 0.18和0.19。可能需要对install script单元格进行一些更改，但在我们的文本打印输出中记录了这些更改。Try it out! :)