Как запустить python скрипт на GPU

Question

В моем блокноте Jupyter torch.cuda.is_available() возвращает True. Но когда я запускаю скрипт python, та же самая строка кода в скрипте python возвращает False. Это говорит мне о том, что когда я запускаю скрипт python в своей записной книжке, он по умолчанию не использует cuda. Любое решение?

Answer 1

Я использовал следующее:

%run -i test_background-matting_image.py -m real-hand-held -i sample_data/input/ -o sample_data/output/ -tb sample_data/background/0001.png

Согласно this : %run -i запускает файл в пространстве имен IPython вместо пустого, то есть "тот же python", как тетрадь моего юпитера.