Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Not Compatible with PyTorch 1.7 + CUDA 11 #118

Open
markyang59 opened this issue Nov 14, 2020 · 3 comments
Open

Not Compatible with PyTorch 1.7 + CUDA 11 #118

markyang59 opened this issue Nov 14, 2020 · 3 comments

Comments

@markyang59
Copy link

Dear
It seems this code is not compatible with
PyTorch 1.7 (lateset now)
Don't have plan for 1.7

@iBobbyTS
Copy link

The latest version or PyTorch you can use for DAIN is 1.4.0, after that, you need a static method for the forward functions, of all the networks. So it's a huge amount of work.
For CUDA 11, did you try if pip install torch==1.4.0 works? If not, you can refer to PyTorch/PyTorch and compile torch 1.4.0 with CUDA 11 by your self.

@flobauer
Copy link

I am currently struggling getting DAIN to work on ubuntu 20.04, gtx3080 and CUDA 11. The mix of getting pytorch 1.4 work with CUDA 11 probably takes me longer than it should.

@laomao0
Copy link

laomao0 commented Nov 7, 2021

If you do not want to build CUDA programs.
We provide the CUPY version of those packages.
The cupy files do not need to be built.
please refer to:
https://github.com/laomao0/cupy_packages

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants