UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error?

重启服务器

参考https://*.com/questions/66371130/cuda-initialization-unexpected-error-from-cudagetdevicecount