-
Notifications
You must be signed in to change notification settings - Fork 254
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
Support for cuda_11.x #106
Comments
why do you use pip to install it? Just use setup.py to install. |
I just followed the instructions given at the main branch. This is what happended:
|
Use pip to install is not support cuda11. It is for old verison. |
Thanks for the advice I cloned the master branch code from i tried editing "/usr/lib/python3/dist-packages/torch/utils/cpp_extension.py" to include CUDA arch (8.6)
I get a new error O.O
|
I think it is no need to edit the cpp_extension.py. That means your PyTorch version is not campatbility with your Cuda. |
Thanks, I tried both the stable and nightly version. This is the error
|
weird... Use command “nvcc -V” to check cuda version. This is the cuda on my machine: |
Yup it is weird haha, I use a RTX 3070
|
I see. You have installed cuda11.2, which is not support by Pytorch currently. You should downgrade to cuda11.1 |
Thanks for your help I installed cuda 11.1 However get this error
Weird, I tried it on both stable and nightly version of pytorch for 11.1
|
Update, thank you @Book4man I was able to solve this by installing 'torch' outside the pip environment
However when I try the examples...
|
hello,did you solve this problem now,i can install it successfuly in pytorch 1.7 1.8 ,but can not install in other low versions,can you help me? |
Why don't you merge this change by asking the fork owner to make PR? What do you say? @adambielski |
@sundoufu I did make a PR 2 years ago #86 |
sorry i didn't look into it well even the comment said it was from PR.... so this repo have not been maintained for a long while |
可以参考这个博客,我在Linux上试了有效:https://blog.csdn.net/axept/article/details/127359947 |
Hello
Thank you for your work. I tried installing it on my machine with RTX-30 but it fails
I already applied changes:
#if CUDA_ARCH < 600 and defined(__CUDA_ARCH__)
for Better Compatibility for CUDA_ARCH for RTX GPUstorch==1.8.1+cu111
match my cuda 11.x/cudnnI get this error
The text was updated successfully, but these errors were encountered: