CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

[lavanvoo]

I receive the following printout when executing simple_test.py from my local machine via the Anaconda prompt:

(pifuhd) C:\Users\Vanforce Productions\Videos\Projects\Digital Doubles\Luke\pifuhd>python -m apps.simple_test Resuming from ./checkpoints/pifuhd.pt Warning: opt is overwritten. test data size: 1 initialize network with normal initialize network with normal generate mesh (test) ... 0%| | 0/1 [00:00<?, ?it/s]./results/pifuhd_final/recon/result_test_512.obj CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle) 100%|████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:22<00:00, 22.70s/it]

This occurs for me even with the test.png file that came with the installation.

Any suggestions for troubleshooting? I am not familiar with the code, but I want to run this PIFuHD tool on my local machine to use more RAM (in order to generate a higher resolution model) than the free Google Collab plan allows. I believe I have installed all required modules.

[egorpezdir]

Same here. Intel i7 9750 2.60GHz NVIDIA GeForce RTX 2060

[Danial-Kord]

Same here, anyone found a solution?

[LongEcho]

Same here

[mtahabahsi]

same here