文章/答案/技术大牛

发布

问如何强制CuPy的CPU使用？
EN

Stack Overflow用户

提问于 2021-10-29 11:37:37

回答 1查看 374关注 0票数 0

这是一个有点奇怪的问题，因为CuPy是为GPU设计的。然而，根据我的程序的输入，我实际上想使用CPU，因为它更快。我已经试过了

if DISABLE_GPU:
  import os
  os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

虽然它阻止了对GPU的访问，但这会导致CuPy抛出一个错误：

cupy_backends.cuda.api.runtime.CUDARuntimeError: cudaErrorNoDevice: no CUDA-capable device is detected

有没有办法强迫CuPy使用CPU而不是GPU？或者我应该用一些技巧比如我说的cp=np ..。

发布于 2022-09-14 05:03:24

CuPy的API是这样的，无论何时使用cp，都是在隐式地使用设备内存。因此，最好的方法是编写代码，以便在需要在CPU上运行的情况下，有条件地使用np而不是cp。

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/69768227

复制

相似问题

问如何强制CuPy的CPU使用？EN