kkvtran

I restarted the travis builds and everything built successfully now.

Hi @Zhaoxian-Wu for the CUDA memory problem, it looked like the problem had to do with how to set the DEVICE. If I set it by `DEVICE = torch.cuda.set_device(0)` instead...

It tried the same technique with torch.empty and I did not see the hanging(looping) issue either. So this is torch problem. Let us know if you have any questions.

Have you tried to following the instructions at this https://aihwkit.readthedocs.io/en/latest/developer_install.html#development-setup and https://aihwkit.readthedocs.io/en/latest/developer_install.html#via-python-command Let me know if you run into issue.