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

Question

我试图在 Google Colab 上安装 cuDF 几个小时。我应该使用 GPU Tesla T4 安装 cuDF 的要求之一。虽然 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.

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

How to install cuDF on google colab with GPU Tesla K80?

dataframe

installation

gpu

google-colaboratory

cudf