0

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

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

所以,我已经切换到可以与 NVIDIA 和 AMD 卡一起使用的 PlaidML。如何在不去终端和使用的情况下更改为 CUP plaidml-setup

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

4

1 回答 1

0

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

于 2022-01-07T08:56:06.827 回答