命令nvidia-smi查看服务器
发现cuda0内存被占满了
改代码 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')为 device = torch.device('cuda:2' if torch.cuda.is_available() else 'cpu')解决
命令nvidia-smi查看服务器
发现cuda0内存被占满了
改代码 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')为 device = torch.device('cuda:2' if torch.cuda.is_available() else 'cpu')解决