RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:411

minar09 / cp-vton-plus

Official implementation for "CP-VTON+: Clothing Shape and Texture Preserving Image-Based Virtual Try-On", CVPRW 2020

https://minar09.github.io/cpvtonplus/

MIT License

343 stars 120 forks source link

RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:411 #77

Open qijindao opened 3 years ago

qijindao commented 3 years ago

When I run the test.py, I met the problem.I have tried many solutions,but they didn't work.My cuda version is 10.1,pytorch is 0.4.1.post2.My PC is RTX2060.I appreciate it greatly if you can help me.

minar09 commented 3 years ago

@qijindao , this is probably your CUDA or driver-related issue. For example: https://github.com/pytorch/pytorch/issues/17334