Question

我正在通过使用nohup进行训练：

nohup python -u main.py &> output.txt &

我已使用以下命令将ipynb转换为python

jupyter nbconvert --to script 'my-notebook.ipynb'

当我在jupyter中运行代码时，它可以正常工作。当我尝试运行python脚本时，我得到了RuntimeError CUDA initialization failed 我同时为jupyter和python分配了PCI_BUS_ID：

try:
    %env CUDA_DEVICE_ORDER=PCI_BUS_ID
    %env CUDA_VISIBLE_DEVICES=5
except NameError:
    os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
    os.environ["CUDA_VISIBLE_DEVICES"] = "5"

我已经完成了2次成功的训练。我稍微修改了一下模型，但现在无法正常工作了。你有什么主意吗？

Answer 1

重新启动PC总是会为我解决此错误

RuntimeError CUDA初始化仅在Python中失败

1 个答案: