How to use GPU for prediction

[eris10]

[mark0428]

To run AMPlify on GPU, you need to install tensorflow-gpu (1.12.0) instead of the default tensorflow.

You may also add some codes like these to the beginning of your script:

import os​
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"