Google Colab：无法安装 cudf

Question

我需要帮助。我正在使用带有 Python 3.10.11 的 Google Colab，并且我有一个带有 CUDA 版本： 12.0 、Nvidia 驱动程序版本 525.85.12 的 Colab，并且我正在按照本教程了解如何安装 cuDF https://colab.research.google.com/驱动器/1TAAi_szMfWqRfHVfjGSqnGVLr_ztzUM9#scrollTo=G_2dancFhf70

我检查了
nvidia-smi

我被分配了 Tesla T4

在台阶上
!python rapidsai-csp-utils/colab/install_rapids.py stable
import os
os.environ\['NUMBAPRO_NVVM'\] = '/usr/local/cuda/nvvm/lib64/libnvvm.so'
os.environ\['NUMBAPRO_LIBDEVICE'\] = '/usr/local/cuda/nvvm/libdevice/'
os.environ\['CONDA_PREFIX'\] = '/usr/local'
!pip uninstall cupy -y

我收到以下警告：
WARNING conda.core.solve:\_add_specs(639): pinned spec python=3.10 conflicts with explicit specs. Overriding pinned spec.
WARNING conda.core.solve:\_add_specs(639): pinned spec cudatoolkit=11.8 conflicts with explicit specs. Overriding pinned spec.

SpecsConfigurationConflictError: Requested specs conflict with configured specs.
requested specs:
- cudatoolkit=11.2
- dask-sql
- gcsfs
- llvmlite
- mamba==1.4.1
- openssl
- python=3.9
- rapids=22.12
pinned specs:
- python_abi=3.10[build=*cp310*]
Use 'conda config --show-sources' to look for 'pinned_specs' and 'track_features'
configuration parameters. Pinned specs may also be defined in the file
/usr/local/conda-meta/pinned.

当我继续下一步时：
import cudf

我收到一个错误：
No module named 'cudf'

有人可以帮我吗？

Answer 1

目前，在 Colab 上安装 RAPIDS 框架的首选方式是使用 pip。如果您使用Rapids.ai/#quick-start上的“启动 Google Colab (Pip)”链接（或直接运行从该链接生成的笔记本中的代码，我在下面列出了这些代码），那么一切都应该适合您。

# This get the RAPIDS-Colab install files and test check your GPU.  Run this and the next cell only.
# Please read the output of this cell.  If your Colab Instance is not RAPIDS compatible, it will warn you and give you remediation steps.
!git clone https://github.com/rapidsai/rapidsai-csp-utils.git
!python rapidsai-csp-utils/colab/pip-install.py