Pytorch"填坑“記(長更)

問題:

存在多個GPU時,指定GPU時的錯誤:RuntimeError: cuda runtime error (10) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cpp:34


解決方法之一:

CUDA_VISIBLE_DEVICES=6(根據自己的GPU個數,任意改變)
./train.py --gpu 0(不要改變它!!!從0開始)

原因之一:

因PyTorch程序設置的gpu_id編號未從0開始導致

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章