You can just add the import ResNet50 or something package imported to use GPU at the top of your code file. That's fine for me.