【发布时间】: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