check for cuda availability

Author: mmz33

i6_models/decoder/attention.py

@@ -130,6 +130,8 @@ def __init__(self, cfg: AttentionLSTMDecoderV1Config):
         self.output = nn.Linear(cfg.output_proj_dim // 2, cfg.vocab_size)
         self.output_dropout = nn.Dropout(cfg.output_dropout)
 
+        if "cuda" in cfg.device:
+            assert torch.cuda.is_available(), "CUDA is not available"
         self.device = cfg.device
 
     def forward(