AssertionError

[pnkj1402]

AssertionError Traceback (most recent call last)
in
82 # Specify the exact model
83 config = gmf_config
---> 84 engine = GMFEngine(config)
85 # config = mlp_config
86 # engine = MLPEngine(config)

1 frames
/content/utils.py in use_cuda(enabled, device_id)
19 def use_cuda(enabled, device_id=0):
20 if enabled:
---> 21 assert torch.cuda.is_available(), 'CUDA is not available'
22 torch.cuda.set_device(device_id)
23

AssertionError: CUDA is not available