我编写了一些深度学习代码,并使用 tensorflow-gpu 库来使用我的 NVIDIA 卡。如果我想使用我的 CPU,我可以写os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
现在,我编写的代码也必须适用于 AMD 卡。
所以,我已经切换到可以与 NVIDIA 和 AMD 卡一起使用的 PlaidML。如何在不去终端和使用的情况下更改为 CUP plaidml-setup
?
我考虑过更改.plaidml
存储配置(GPU 或 CPU)的文件,但没有成功。