import torch
print(torch._C._cuda_getDeviceCount())
<stdin>:1: 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.)
这是因为显卡cuda, 驱动没装好的原因, 重装就好了.