开始学习pytorch,看到一行代码 assert torch.cuda.is_available()
了解一下assert的用法
assert 声明,主张
作用:检查程序,不符合条件即终止程序
- a=3
- assert a>4 #报错
- assert a<4 #运行成功
- assert not torch.cuda.is_available() #运行成功(因为这台电脑没gpu,所以没部署cudu, cuDNN)
开始学习pytorch,看到一行代码 assert torch.cuda.is_available()
了解一下assert的用法
assert 声明,主张
作用:检查程序,不符合条件即终止程序
- a=3
- assert a>4 #报错
- assert a<4 #运行成功
- assert not torch.cuda.is_available() #运行成功(因为这台电脑没gpu,所以没部署cudu, cuDNN)