import torch>
>>> print(torch.version) # Should show 2.2.0
2.2.0+cu121
>>> print(torch.version.cuda) # Should show 12.1
12.1
>>> print(torch.cuda.is_available()) # Should return True
/usr/local/lib/python3.10/dist-packages/torch/cuda/init.py:141: UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 803: system has unsupported display driver / cuda driver combination (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:108.)
return torch._C._cuda_getDeviceCount() > 0
False
>>> exit()