這個(gè)問(wèn)題在不同的運(yùn)行環(huán)境下有不同的解決方案。
一種可行的解決方案:在代碼中設(shè)置CUDA_VISIBLE_DEVICES
環(huán)境變量后,調(diào)用torch.cuda.device_count.cache_clear()
宣鄙,例如:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "1"
import torch
torch.cuda.device_count.cache_clear()