解決運行出現CUDA error:out of memory的問題

模型運行出現以下問題,經過查找資料發現解決方案如下: CUDA_VISIBLE_DEVICES限制一下使用的GPU。 比如有0,1,2,3號GPU,CUDA_VISIBLE_DEVICES=2,3,則當前進程的可見GPU只有物理上的2、3號GPU,此時它們的編號也對應變成了0、1,即cuda:0對應2號GPU,cuda:1對應3號GPU。 如何設置CUDA_VISIBLE_DEVICES: ① 使
相關文章
相關標籤/搜索