[ 组合替代实现 ]torch.cuda.is_available¶
torch.cuda.is_available¶
torch.cuda.is_available()
转写示例¶
# torch 写法
torch.cuda.is_available()
# paddle 写法
paddle.device.cuda.device_count() >= 1
torch.cuda.is_available()
# torch 写法
torch.cuda.is_available()
# paddle 写法
paddle.device.cuda.device_count() >= 1