torch.cuda.is_available()
torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
torch.cuda.get_device_name(0)
nvidia-smi
参考链接
NVIDIA pytorch命令查看
最新推荐文章于 2024-03-22 16:18:39 发布
torch.cuda.is_available()
torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
torch.cuda.get_device_name(0)
nvidia-smi