gpt4 book ai didi

linux - OpenCL 不显示 Nvidia GPU

转载 作者:塔克拉玛干 更新时间:2023-11-02 23:43:48 29 4
gpt4 key购买 nike

希望这不是重复的,我确实环顾四周但如果我错了请纠正我。

我最近安装了 pyopencl通过 Anaconda,它可以毫无问题地在我的 Intel CPU 上运行。但是,当我打电话时:

for platform in get_platforms():
print(platform.name)

ctx = create_some_context()

for dev in ctx.devices:
print(dev.name)

我得到的只是:

Portable Computing Language
Choose platform:
[0] <pyopencl.Platform 'Portable Computing Language' at 0x7f0ff2205020>
>> 0
Choice [0]:0
Set the environment variable PYOPENCL_CTX='0' to avoid being asked again.
pthread-Intel(R) Core(TM) i7-7700HQ CPU @ 2.80GHz

我有 Nvidia GeForce GTX 1050 Ti,驱动程序版本是 390.87,带有 CUDA 9.1,根据 Nvidia 发行说明应该是兼容的。我也在运行 Linux Mint 19.1(如果相关的话),并且在涉及到这些东西时我也有点菜鸟,因为我的背景是物理学,而不是计算机科学。

我在某处(我不记得在哪里)读到我应该有一个 nvidia.icd 文件,但现在只有/home//anaconda3/etc/OpenCL/vendors 中的一个 pocl.icd 文件,我没有确定如何/在何处获取 nvidia.icd 文件。

在此先感谢您的帮助,如果您需要更多信息,请告诉我!

编辑:添加了 get_platforms() 的结果

编辑 2:我找到了一个 nvidia.icd文件在我的/etc/OpenCL/vendors目录。我将其复制到 Anaconda 目录,/home/<user>/anaconda3/etc/OpenCL/vendorspyopencl现在识别Nvidia CUDA作为平台,我的 GPU 作为设备。

不幸的是,它仍在将任务加载到 CPU,但这很可能是因为我的代码,我还没有正确地弄清楚如何在 GPU 上创建上下文。

最佳答案

这个问题有点陈旧,但我在尝试准确解决这个问题时遇到了它,我在 Ubuntu Linux 的 Conda 环境中有一个新安装的 PyOpenCL 包,它没有在列表中列出我的 GPU平台及其相关设备。

OP 的第二次编辑,他从 /etc/OpenCL/vendors 复制了 nvidia.icd 文件对我有用,但感觉不对,所以我挖了一下深入 PyOpenCL documentation ,它建议,除了 PyOpenCL 包之外,执行 conda install ocl-icd-system。 (请参阅该页面上的“使用供应商提供的驱动程序”部分。)

这行得通,并为您提供了所有系统功能——我认为这是一种比到处复制 .icd 文件更简洁的方法。

关于linux - OpenCL 不显示 Nvidia GPU,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54294760/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com