torch cuda is available
torch.cuda.is_available() torch.cuda.current_device() torch.cuda.device(0) torch.cuda.device_count() torch.cuda.get_device_name(0)
Source: stackoverflow.com
how to install pytorch 0.4.1
pip install --trusted-host pypi.org --trusted-host files.pythonhosted.org torch===0.4.1 torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html
torch cuda version
print(torch.version.cuda)
Source: discuss.pytorch.org