【问题标题】:PlaidML, switching between cpu and gpuPlaidML,cpu和gpu之间的切换
【发布时间】:2021-12-29 19:18:14
【问题描述】:

我编写了一些深度学习代码,并使用 tensorflow-gpu 库来使用我的 NVIDIA 卡。如果我想使用我的 CPU,我可以写 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

现在,我编写的代码也必须适用于 AMD 卡。

所以,我已切换到可与 NVIDIA 和 AMD 卡一起使用的 PlaidML。不去终端使用plaidml-setup如何换成CUP?

我想过更改存储配置(GPU 或 CPU)的文件.plaidml,但没有成功。

【问题讨论】:

    标签: tensorflow amd plaidml


    【解决方案1】:

    在plaidml的库中,有一个名为plaidmlsetup的python文件,可以用来改变cpu和gpu。

    【讨论】:

    • 正如目前所写,您的答案尚不清楚。请edit 添加其他详细信息,以帮助其他人了解这如何解决所提出的问题。你可以找到更多关于如何写好答案的信息in the help center
    猜你喜欢
    • 1970-01-01
    • 2012-03-07
    • 2013-09-27
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2018-12-16
    • 1970-01-01
    相关资源
    最近更新 更多