如何使用 GPU Tesla K80 在 google colab 上安装 cuDF？

Question

我尝试在 Google Colab 上安装 cuDF 几个小时。我应该安装 cuDF 和 GPU Tesla T4 的要求之一。虽然 google colab 每次都给我 GPU Tesla K80 但我无法安装 cuDF。我每次都尝试使用这段代码来检查我拥有的 GPU 类型：

import pynvml

pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(0)
device_name = pynvml.nvmlDeviceGetName(handle)

if device_name != b'Tesla T4':
  raise Exception("""
    Unfortunately this instance does not have a T4 GPU.
    
    Please make sure you've configured Colab to request a GPU instance type.
    
    Sometimes Colab allocates a Tesla K80 instead of a T4. Resetting the instance.

    If you get a K80 GPU, try Runtime -> Reset all runtimes...
  """)
else:
  print('Woo! You got the right kind of GPU!') 

通过 google colab 获取特定类型的 GPU 太令人沮丧了，因为这有点运气。想请问一下有没有人遇到同样的问题，是怎么解决的？

Answer 1

K80使用Kepler GPU架构，RAPIDS不支持该架构。 Colab 本身不再可以运行最新版本的 RAPIDS。您可以尝试 SageMaker Studio Lab 来体验“立即尝试”。 https://github.com/rapidsai-community/rapids-smsl。