是否可以使用conda或pip将pyarrow与cuda集成安装?

时间:2019-08-02 14:53:42

标签: pip conda pyarrow

我正在尝试使用pyarrow.cuda。但是通过pyarrowconda安装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,结果是相同的。

0 个答案:

没有答案