import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu' )
切换gpu显卡
猜你喜欢
转载自blog.csdn.net/weixin_43722052/article/details/132910030
今日推荐
周排行