NVIDIA-Merlin/core

[Bug] Installing latest branch of Merlin core raises error when importing cuDF

bschifferer opened this issue · 0 comments

bschifferer commented

When I pull the latest branch of Merlin core and install it, I cannot import cuDF anymore.

Steps to reproduce:

  • docker pull merlin-training-tensorflow:22.04
  • cd /core/ && git pull origin main && pip install .
  • python
  • import cuDF

Error:

File cudf/_lib/gpuarrow.pyx:1, in init cudf._lib.gpuarrow()

ModuleNotFoundError: No module named 'pyarrow._cuda'