0

我试图在 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 太令人沮丧了，因为这是一种运气。我在这里问是否有人遇到过同样的问题，以及如何解决的？

4

0 回答 0