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开始导致

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