No it will work with AMD. You will have to build it with pytorch thats based on AMD Radeon open compute (RocM).
