Closed taylover-pei closed 5 years ago
As a quick sanity check, could you run python --version
and let me know what it prints?
I got the same error as well. It seems to happen when cupy is not installed correctly.
I solved it by using pip install cupy-cuda90
to install cupy (follow the instruction of https://github.com/cupy/cupy), with Python 3.6.6 and Pytorch 0.4.1.
When I try to run your code, errors happened like this:
I uese python3.6, pytorch0.4。So, what should I do to solve the problem. Looking forward to your reply. Thank you very much.