latent-diffusion
latent-diffusion copied to clipboard
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling 'cublasCreate(handle)'
I want to test the 'tex2img.py', but has this Error; anyone know how to fix this? thanks~ using TATIAN RTX 24G, pytorch 1.7.0, torchvision 0.8.1, cudatoolkit 11.0.3.
(Forgive me, I don't have an internet connection, I can only take pictures with my phone)
/pytorch/aten/src/Aten/native/cuda/Indexing.cu"658: indexSelectLargeIndex: block: [31,0,0], thread: [127,0,0] Asserting 'srcIndex
< srcSelectDimSize' failed.
