Leo Liu

I also come across with the problem. And it is because pytorch is cpu-version, so run `conda install -c pytorch pytorch` can solve the problem.