pytroch使用gpu命令torch.cuda简单总结
pytorch的gpu使用常用的和gpu有关的命令torch.cuda.is_available() # 判断是否可以使用gpu# 设置能使用的gpu号，但这里设置有，但实际gpu没这个编号，在使用时会报错os.environ['CUDA_VISIBLE_DEVICES'] = '0,3' # 要放在import torch之前torch.cuda.device_cou
