In [2]: torch.__version__
Out[2]: '0.4.0'

In [3]: torch.device("cuda")
Out[3]: device(type='cuda')

In [4]: cuda = torch.device("cuda")

In [5]: torch.tensor([[1], [2], [3]], dtype=torch.half, device=cuda)
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1524584710464/work/aten/src/THC/THCGeneral.cpp line=70 error=30 : unknown error
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-5-6f6a0c1a1f8d> in <module>()
----> 1 torch.tensor([[1], [2], [3]], dtype=torch.half, device=cuda)

~/miniconda3/envs/torch/lib/python3.6/site-packages/torch/cuda/__init__.py in _lazy_init()
    159             "Cannot re-initialize CUDA in forked subprocess. " + msg)
    160     _check_driver()
--> 161     torch._C._cuda_init()
    162     _cudart = _load_cudart()
    163     _cudart.cudaGetErrorName.restype = ctypes.c_char_p

RuntimeError: cuda runtime error (30) : unknown error at /opt/conda/conda-bld/pytorch_1524584710464/work/aten/src/THC/THCGeneral.cpp:70

In [6]: torch.cuda.is_available()
Out[6]: False