問題:
存在多個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開始導致