-
Notifications
You must be signed in to change notification settings - Fork 66
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
pytorch1.6 cuda10.2 #5
Comments
Yes. I did it. First install PyTorch 1.6.0 (best in a virtual env) and CUDA 10.2. Make the Then, from the mish-cuda repository, run Hope this helps. |
Just setting Also you can just use |
Hi, I use torch=1.6. I use "pip install git+https://github.com/thomasbrandon/mish-cuda/" to install mish-cuda. But it give me an error like: and How can I solve them? Thanks ! |
@thomasbrandon |
@changxinC This is related to an incompatability between your system C++ compiler and your CUDA version, It's apparently fixed in CUDA 10.1.168 (according to this). Otherwise upgrading/downgrading your system C++ compiler may fix. Exactly which is easiest will depend a on your distro. |
Oh thanks, my CUDA is 10.1.105, g++ is 7.6.5, maybe I need to update the CUDA version. |
I got the same problem. Not sure if you have resolved it? Thanks. |
Instead of installing it, I just defined a class that implements mish by pytorch. |
Thank you. I figured the same thing lol |
pytorch1.6 cuda10.2
Can this be installed?
The text was updated successfully, but these errors were encountered: