我已经安装了pytorch，想检查是否有任何脚本来测试安装是否正确，例如是否可以启用CUDA等？
deep-learning pytorch
我知道我可以使用 访问当前的 GPU torch.cuda.current_device()，但是如何获取所有当前可用 GPU 的列表？
torch.cuda.current_device()
python gpu pytorch
pytorch ×2
deep-learning ×1
gpu ×1
python ×1