gpt4 book ai didi

python - 无法在纯 CPU Caffe 中使用 GPU : check mode

转载 作者:太空宇宙 更新时间:2023-11-03 16:00:17 27 4
gpt4 key购买 nike

我正在使用在 CPU 上运行的旧版 caffe 设置。

我切换了一行代码:

caffe.set_mode_cpu()

至:

caffe.set_mode_gpu()

但我收到错误:

Cannot use GPU in CPU-only Caffe: check mode.

我检查了Makefile.config,相关设置是:

#CPU-only switch (uncomment to build without GPU support)
CPU_only := 1

我假设 1 是 True,所以我改为

CPU_only := 0

但这并没有帮助。

有什么线索吗?

最佳答案

您需要更改:

#CPU-only switch (uncomment to build without GPU support)
CPU_only := 1

对此:

#CPU-only switch (uncomment to build without GPU support)
#CPU_only := 1

并重建。

关于python - 无法在纯 CPU Caffe 中使用 GPU : check mode,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40382361/

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