#!/usr/bin/env python
"""Checks that PyTorch can reach CUDA."""

import sys

import torch


if __name__ == "__main__":
    if not torch.cuda.is_available():
        print("FAIL", file=sys.stderr)
        exit(1)
    else:
        print("OK", file=sys.stderr)