I have nvidia 1080TI, Ubuntu x64 and Python 3.6.9 installed. I was trying to launch pytorch with command
import torch print(torch.cuda.is_available)