Google Colab安装cuDF遇版本冲突及模块缺失问题求助

问题原因 #

• 你使用的stable版RAPIDS安装脚本默认依赖较低版本的Python（如3.9）和CUDA（如11.2），但当前Colab环境是Python 3.10.11、CUDA 12.0，版本不匹配触发了SpecsConfigurationConflictError，导致cuDF安装失败。
• 安装脚本内的固定依赖python_abi=3.10[build=*cp310*]与请求的低版本Python/CUDA冲突，conda强制覆盖依赖后仍无法完成兼容安装，最终导致import cudf时报错No module named 'cudf'。

解决方案 #

1. 安装适配CUDA 12.0的RAPIDS版本 #

清理之前的安装残留，改用支持CUDA 12.0的RAPIDS版本（23.10及以后版本已支持CUDA 12），执行以下命令：

# 清理残留的相关包
!pip uninstall -y cupy cudf
!conda remove -y rapidsai

# 安装适配CUDA 12.0的stable版RAPIDS
!python rapidsai-csp-utils/colab/install_rapids.py stable --cuda-version=12.0

2. 配置正确的环境变量 #

安装完成后，重新设置匹配CUDA 12.0的环境变量：

import os
os.environ['NUMBAPRO_NVVM'] = '/usr/local/cuda-12.0/nvvm/lib64/libnvvm.so'
os.environ['NUMBAPRO_LIBDEVICE'] = '/usr/local/cuda-12.0/nvvm/libdevice/'
os.environ['CONDA_PREFIX'] = '/usr/local'

3. 验证安装结果 #

运行以下代码确认cuDF安装成功：

import cudf
print(cudf.__version__)

4. 备选方案：切换Colab环境版本 #

如果上述方法无效，可以手动切换Colab到兼容的环境：

• 点击顶部菜单栏Runtime -> Change runtime type，选择Python 3.9（若可选），并保持硬件加速器为GPU。
• 重新运行原教程的安装脚本，此时版本匹配，不会触发依赖冲突。

内容的提问来源于stack exchange，提问作者Nata107