Pytorch 1.7 compatibility about aihwkit HOT 3 CLOSED

In regards to using the current 0.2.0 wheels:

$ pip freeze
aihwkit==0.2.0
torch==1.7.0

This combination seem to result in:

$ python -c "import aihwkit.simulator.rpu_base"
Traceback (most recent call last):
  File "<string>", line 1, in <module>
ImportError: /tmp/.../lib/python3.7/site-packages/aihwkit/simulator/rpu_base.cpython-37m-x86_64-linux-gnu.so:
undefined symbol: _ZN3c104impl23ExcludeDispatchKeyGuardC1ENS_11DispatchKeyE

As a workaround while troubleshooting the issue and working on a cleaner solution, manually installing an earlier version of torch should still do the trick:

pip install "torch<1.7"

from aihwkit.

An update regarding torch compatibility - we just issued a 0.2.1 release that contains wheels that are compatible with torch 1.6.0. In the next major release 0.3.0 we are planning to update the wheels dependency to 1.7.x.

Please note that this compatibility refers only to the pip-installable wheels: when they are built, our library is compiled against a specific torch version, and the same specific torch versions needs to be installed in your system. For that reason, in the 0.2.1 we pinned the dependency on pytorch: when installing aihwkit, it will force the installation of torch 1.6.0 in your virtual environment.

Using a self-compiled version of aihwkit (as in, not the wheels from pip, but the ones that can be produced via the advanced install instructions) along with other version of pytorch (including 1.7) should be possible - the codebase is compatible with 1.6 and 1.7.

from aihwkit.

Closing in favour of #204

from aihwkit.