import torch


def cuda_network(network):
    """Converts network to CUDA if available"""
    if torch.cuda.is_available():
        print('CUDA available: converting network to CUDA')
        network.cuda()
    return network