RuntimeError: cuda runtime error (11) : invalid argument at /opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/THC/THCGeneral.cpp:663 - Githubissues

CuriousAI / mean-teacher

A state-of-the-art semi-supervised method for image recognition

https://arxiv.org/abs/1703.01780

Other

1.56k stars 331 forks source link

RuntimeError: cuda runtime error (11) : invalid argument at /opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/THC/THCGeneral.cpp:663 #41

Open bolin12 opened 4 years ago

bolin12 commented 4 years ago

python 3.6 cuda9.0 pytorch0.4