开头加上
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘0,1,2,3’
改为:
device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’ )
RuntimeError: Invalid device string: ‘0,1,2,3‘
于 2022-05-23 16:22:07 首次发布
开头加上
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘0,1,2,3’
改为:
device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’ )