我正在使用在 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/