RSN 20 - 请将 cuda-python 固定到 11.7.0 版本

作者 RAPIDS Ops
状态

已完成

主题 平台支持变更
RAPIDS 版本 v22.02、v22.04 和 v22.06
创建日期 2022 年 7 月 15 日
更新日期 2023 年 9 月 5 日

概述

我们最近发现 cuda-python 11.7.1 引入了一些变更,导致在尝试安装 RAPIDS 时与当前和旧版本的 RAPIDS 不兼容。您需要在 conda 安装期间,以及可能在安装后,将 cuda-python 固定到 11.7.0 版本。受 11.7.1 问题影响的安装会导致抛出此错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/opt/conda/lib/python3.9/site-packages/cudf/__init__.py", line 5, in <module>
    validate_setup()
  File "/opt/conda/lib/python3.9/site-packages/cudf/utils/gpu_utils.py", line 20, in validate_setup
    from rmm._cuda.gpu import (
  File "/opt/conda/lib/python3.9/site-packages/rmm/__init__.py", line 16, in <module>
    from rmm import mr
  File "/opt/conda/lib/python3.9/site-packages/rmm/mr.py", line 14, in <module>
    from rmm._lib.memory_resource import (
  File "/opt/conda/lib/python3.9/site-packages/rmm/_lib/__init__.py", line 15, in <module>
    from .device_buffer import DeviceBuffer
  File "device_buffer.pyx", line 1, in init rmm._lib.device_buffer
TypeError: C function cuda.ccudart.cudaStreamSynchronize has wrong signature (expected __pyx_t_4cuda_7ccudart_cudaError_t (__pyx_t_4cuda_7ccudart_cudaStream_t), got cudaError_t (cudaStream_t)

状态

目前,我们提供以下两种解决途径:

Resolution Path 1: You can upgrade your RAPIDS version to `v22.06.01`.
- We've made it easy to upgrade: simply use our Release selector and install `v22.06` or rerun you `v22.06` installation script. It will automatically grab `v22.06.01` via Conda or Docker.

Resolution Path 2: If you can NOT upgrade, pin `cuda-python` to `11.7.0` in RAPIDS `v22.02`, `v22.04`, and `v22.06`,
- You can pin `cuda-python` to `11.7.0` by using `cuda-python=11.7.0` when doing an install.
- Example command: `conda install -c nvidia cuda-python=11.7.0`.

较旧版本的 RAPIDS 也可能受到影响,用户应尝试升级或固定版本。

影响

此问题影响安装依赖于 cuda-python 的 RAPIDS 22.02 至 22.06 版本 conda 包的用户。如果用户安装了依赖于 cuda-python 的额外包,导致 cuda-python 更新到 11.7.0 以上版本,此问题也会影响现有的 RAPIDS 22.02 - 22.06 版本 conda 安装。