解決運行出現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: ① 使

>>阅读原文<<