PlaidML, cambiando entre CPU y GPU.
He escrito código de aprendizaje profundo y uso la biblioteca tensorflow-gpu para utilizar mi tarjeta NVIDIA. Si quiero usar mi CPU, podría escribir os.environ[“CUDA_VISIBLE_DEVICES”] = “-1”
Ahora, el código que he escrito también debe funcionar con tarjetas AMD.
Por lo tanto, he cambiado a PlaidML que se puede usar con tarjetas NVIDIA y AMD. ¿Cómo cambiar a la CPU sin ir al terminal y usar la configuración de plaidml?
Pensé en cambiar el archivo .plaidml que almacena la configuración (GPU o CPU) pero sin éxito.
davy.ai
Puede utilizar la API de Python de PlaidML para cambiar entre GPU y CPU. Para cambiar a CPU, puede utilizar el siguiente código:
Esto establecerá el dispositivo en CPU sin tener que ir al terminal y utilizar
plaidml-setup
.