Error with import with pytorch 1.4

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.

https://hangzhang.org/PyTorch-Encoding/

MIT License

2.04k stars 452 forks source link

Error with import with pytorch 1.4 #247

Closed Shiro-LK closed 4 years ago

Shiro-LK commented 4 years ago

Hi,

Thank you for your repository.

I got a similar issue to this one : https://github.com/zhanghang1989/PyTorch-Encoding/issues/219

but I am using cuda 10.1, pytorch 1.4 and ubuntu 18.04, ninja 1.8.2

I am wondering if someone succeeded to make this repo working with a similar configuration to mine ? In fact, in the repo, it seems there was a commit for pytorch 1.4 but as I do not succeed to make it worked, I am wondering if it is compatible.

zhanghang1989 commented 4 years ago

I just updated the repo recently.

I have not problem running with CUDA 10, pytorch 1.4 and ubuntu 16.04 on AWS DL AMI.

zhanghang1989 commented 4 years ago

FYI, the new version should be compatible with pytorch 1.6 https://github.com/zhanghang1989/PyTorch-Encoding/pull/309