如何在gpu上运行python脚本

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 的命名空间中运行文件，而不是一个空文件，即与我的 jupyter 笔记本“相同的 python”。