Windows10下安装Rapids.ai遇包找不到错误，求安装指导

一、先确认WSL2环境的前置要求 #

• 确保WSL2里的Ubuntu发行版是20.04或22.04（RAPIDS对这两个版本支持最完善）
• 检查CUDA版本：打开WSL终端，运行nvidia-smi，确认CUDA版本为11.8或12.0（RAPIDS 23.10仅适配这两个CUDA版本，版本不匹配直接导致找不到对应包）
• 若CUDA版本不对，得先在WSL2里安装对应版本的CUDA Toolkit，别用apt默认源的版本，要装NVIDIA官方提供的对应版本

二、conda安装（官方推荐） #

之前找不到包大概率是源配置不对或版本不匹配，按以下步骤来：

1. 更新conda（没装conda的话先装Miniconda/Anaconda）：

   conda update -n base conda -y
   

2. 创建并激活专属环境，指定Python版本（RAPIDS 23.10支持3.8-3.10）：

   conda create -n rapids-23.10 python=3.10 -y
   conda activate rapids-23.10
   

3. 添加正确的conda源：

   conda config --add channels conda-forge
   conda config --add channels nvidia
   conda config --add channels rapidsai
   

4. 按CUDA版本安装RAPIDS：
   • CUDA 11.8用户执行：

     conda install -c rapidsai -c nvidia -c conda-forge rapids=23.10 python=3.10 cudatoolkit=11.8
     

   • CUDA 12.0用户执行：

     conda install -c rapidsai -c nvidia -c conda-forge rapids=23.10 python=3.10 cudatoolkit=12.0
     

5. 验证安装：

   python -c "import cudf; print(cudf.__version__)"
   

   输出23.10.x即为成功

三、pip安装（备选方案） #

pip安装的包命名规则和conda不同，之前直接搜rapids=23.10*找不到是正常的，按以下步骤操作：

1. 更新pip到最新版：

   python -m pip install --upgrade pip
   

2. 按CUDA版本安装：
   • CUDA 11.8用户执行：

     pip install rapids-cu118==23.10.* --extra-index-url=https://pypi.nvidia.com
     

   • CUDA 12.0用户执行：

     pip install rapids-cu120==23.10.* --extra-index-url=https://pypi.nvidia.com
     

   注意：RAPIDS的pip包托管在NVIDIA专属PyPI源，必须加--extra-index-url参数才能找到

四、排查要点 #

如果还是找不到包，检查这几点：

• WSL2的Ubuntu版本是否是20.04/22.04，其他版本没有预编译的RAPIDS包
• CUDA版本是否严格对应11.8或12.0，版本差一点都不行
• conda用户可以运行conda config --show channels，确认rapidsai、nvidia、conda-forge这三个源都在列表里
• pip用户确认加了NVIDIA的PyPI源，默认PyPI源没有RAPIDS包

内容的提问来源于stack exchange，提问作者John David Deatherage