RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
jlkokko opened this issue · 1 comments
jlkokko commented
Cuda shows as installed and available by torch. Imagine runs ok until I pass any parameters (ubuntu wsl2).
site-packages/torch/autograd/init.py", line 173, in backward
Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
jlkokko commented
Appears to be a parameter issue I've worked out...