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

Question

我正在尝试使用

pyarrow.cuda

。但是通过

pyarrow

或

conda

安装

pip

后，出现以下错误：

>>> 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

环境，通过以下方式创建：

$ 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'

我还创建了一个新的

conda

环境，其中

pyarrow

是从

pip

安装的，结果是相同的。

Answer 1

conda 安装 pyarrow“arrow-cpp-proc=*=cuda”

我还不知道如何使用 pip 来做到这一点。