是否可以使用conda或pip安装带有cuda集成的pyarrow？

我正在尝试使用

pyarrow.cuda

。但是在通过

conda

或

pip

安装

pyarrow

后，我得到以下错误：

>>> import pyarrow.cuda
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/b1es/miniconda3/envs/seismic/lib/python3.6/site-packages/pyarrow/cuda.py", line 22, in <module>
    from pyarrow._cuda import (Context, IpcMemHandle, CudaBuffer,
ModuleNotFoundError: No module named 'pyarrow._cuda'

---

我还创建了一个新的

conda

env，其中

pyarrow

是从

pip

安装的，结果是一样的。

按照您当前的提问方式，不可能知道您是如何设置该环境的。请仅使用

conda

从一个全新的环境开始。您是否已使用conda安装CUDA？还是有一个pyarrow构建依赖于CUDA？尝试使用

pip

时，请使用另一个单独的环境。@RolandWeber我编辑了我的问题。你问我是不是用康达安装了CUDA是什么意思？CUDA Toolkit是在miniconda.conda拥有自己的CUDA Toolkit包之前安装的。如果您在conda之外安装了工具包，则可能无法自动找到它。不幸的是，这没有帮助。首先我创建了环境，然后我安装了

cudatoolkit

，然后安装了

pyarrow

。按照您当前的提问方式，无法确定您是如何设置该环境的。请仅使用

conda

从一个全新的环境开始。您是否已使用conda安装CUDA？还是有一个pyarrow构建依赖于CUDA？尝试使用

pip

时，请使用另一个单独的环境。@RolandWeber我编辑了我的问题。你问我是不是用康达安装了CUDA是什么意思？CUDA Toolkit是在miniconda.conda拥有自己的CUDA Toolkit包之前安装的。如果您在conda之外安装了工具包，则可能无法自动找到它。不幸的是，这没有帮助。首先我创建了环境，然后安装了

cudatoolkit

，然后安装了

pyarrow

。

$ conda create -n seismic -c conda-forge python numpy scipy sympy matplotlib numba pandas pillow pyspark tqdm pyarrow
...
$ conda activate seismic
$ python -c 'from numba import cuda ; print(cuda.current_context())'
<CUDA context c_void_p(93844987691120) of device 0>
$ python -c 'import pyarrow ; print(pyarrow.__version__)'
0.14.1
$ python -c 'import pyarrow.cuda'
Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/home/b1es/miniconda3/envs/seismic/lib/python3.7/site-packages/pyarrow/cuda.py", line 22, in <module>
    from pyarrow._cuda import (Context, IpcMemHandle, CudaBuffer,
ModuleNotFoundError: No module named 'pyarrow._cuda'