Open alessiapacca opened 3 years ago
I tried to run the single-image demonstration on one image, using pytorch 1.0.0 and cud 10.0, but it gives me this output:
pre-trained model is loaded step:3, alpha:1 iteration:80000 THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1544174967633/work/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument
I mean, I am using pytorch 1.0.0 as you are, so I don't see where the problem could be. Any idea?
Hi, alessiapacca. Thanks for your interest out work.
Please refer the link.
I tried to run the single-image demonstration on one image, using pytorch 1.0.0 and cud 10.0, but it gives me this output:
I mean, I am using pytorch 1.0.0 as you are, so I don't see where the problem could be. Any idea?