python代码查看可用的gpu,然后指定可用的gpu运行
1.查看gpu (最好像下面这样代码里输出看下,nvidia-smi里gpu编号有时候对不上的) import torch # 检查CUDA是否可用 if torch.cuda.is_available(): # 获取GPU设备数量 dev...
1.查看gpu (最好像下面这样代码里输出看下,nvidia-smi里gpu编号有时候对不上的) import torch # 检查CUDA是否可用 if torch.cuda.is_available(): # 获取GPU设备数量 dev...