如何使用pip安装cudf?

6
我想要加速我在GPU上的pandas操作,所以我决定使用cudf库。如果有其他的库,请建议。
我尝试使用pip安装cudf,命令是pip3.6 install cudf-cuda92。pip版本为19.2.3(最新版)。
当我在cmd上运行pip3.6 install cudf-cuda92时,它会显示:
Collecting cudf-cuda92
  ERROR: Could not find a version that satisfies the requirement cudf-cuda92 (from versions: none)
ERROR: No matching distribution found for cudf-cuda92

你的操作系统是什么?cudf 只能在 Linux 上使用。 - FlyingTeller
5个回答

2
获取您的GPU的cuda支持版本
nvidia-smi

enter image description here

https://docs.rapids.ai/install获取支持的cuda版本的下载命令。

This is where you select the requirments to get the command

3. 复制该命令并执行。

2

cudf 的pip安装官方不被支持也不再维护。请查看 cudf github 上的讨论和官方 rapids 公告

推荐使用 conda 或通过源代码构建。


2
您可以使用pip安装,但请注意cuda版本的依赖性可能相当严格,并且可能需要与cudf版本匹配。为此提供了不同的cudf版本,但它们可能落后于最新版。
例如,我正在运行cuda 10.0,因此安装了cudf:
pip3 install cudf-cuda100

然后在 /etc/ld.so.conf.d 中添加一个文件,指向 /usr/local/cuda-10.0/lib64。

(我正在运行 Ubuntu 18.04)


2

你可以尝试使用conda进行安装。你可以通过Miniconda获取一个最小的conda安装包,或者通过Anaconda获取完整的安装包。

使用conda命令安装和更新cuDF:

# CUDA 9.2
conda install -c nvidia -c rapidsai -c numba -c conda-forge -c defaults cudf

# CUDA 10.0
conda install -c nvidia/label/cuda10.0 -c rapidsai/label/cuda10.0 -c numba -c conda-forge -c defaults cudf

请访问cudf.PyPi了解更多关于it技术的信息。


0

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接