WSL2安装RAPIDS后导入cudf遇ModuleNotFoundError求助

问题背景 #

已在WSL2环境完成RAPIDS 24.06安装（执行conda创建环境命令时提示环境已存在），但导入cudf时出现ModuleNotFoundError。

操作日志 #

(base) aridleprechaun@Samuel-Tanner-Desktop:~$ ls
Anaconda3-2024.06-1-Linux-x86_64.sh    Anaconda3-2024.06-1-Windows-x86_64.exe    anaconda3
Anaconda3-2024.06-1-Linux-x86_64.sh.1  Anaconda3-2024.06-1-Windows-x86_64.exe.1
(base) aridleprechaun@Samuel-Tanner-Desktop:~$ conda create -n rapids-24.06 -c rapidsai -c conda-forge -c nvidia  \
    rapids=24.06 python=3.11 'cuda-version>=12.0,<=12.2'
WARNING: A conda environment already exists at '/home/aridleprechaun/anaconda3/envs/rapids-24.06'
Remove existing environment (y/[n])? n


CondaSystemExit: Exiting.

(base) aridleprechaun@Samuel-Tanner-Desktop:~$ python
Python 3.12.4 | packaged by Anaconda, Inc. | (main, Jun 18 2024, 15:12:24) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import cudf
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'cudf'
>>> import pandas

解决方案 #

1. 激活RAPIDS专属conda环境 #

从日志可见，你当前处于base环境（命令行前缀显示(base)），而cudf安装在rapids-24.06环境中，必须先激活该环境：

conda activate rapids-24.06

激活后命令行前缀会变为(rapids-24.06)，此时再启动Python并导入cudf即可。

2. 验证环境内的cudf安装状态 #

如果激活环境后仍报错，先检查环境内是否已安装cudf：

conda activate rapids-24.06
conda list | grep cudf

若无输出，说明环境内未正确安装cudf，重新执行安装命令：

conda activate rapids-24.06
conda install -c rapidsai -c conda-forge -c nvidia cudf=24.06

3. 确认Python版本匹配 #

你创建RAPIDS环境时指定的是Python 3.11，但base环境的Python是3.12.4。激活rapids-24.06后执行python --version，确认版本为3.11.x，版本不匹配会导致包无法识别。

4. 修复损坏的conda环境（可选） #

若环境存在损坏，可尝试恢复到初始状态后重新安装：

conda activate rapids-24.06
conda install --revision 0

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