Failed inference for CodeFormer: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`

[xingguo-Zhang]

When I run the following code, python inference_codeformer.py -w 0.5 --has_aligned --input_path inputs/cropped_faces

I get the following error, can anyone tell me the possible cause?

Background upsampling: False, Face upsampling: False [1/2] Processing: 0143.png Failed inference for CodeFormer: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle) [2/2] Processing: 0240.png Failed inference for CodeFormer: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle)