-
Notifications
You must be signed in to change notification settings - Fork 145
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
Code works on one GPU machine, doesn't work on another #36
Comments
File "/home/xxx/anaconda3/lib/python3.7/site-packages/torch_cluster/knn.py", line 123, in knn_graph File "/home/xxx/anaconda3/lib/python3.7/site-packages/torch_cluster/knn.py", line 60, in knn RuntimeError: CUDA error: no kernel image is available for execution on the device (launch_kernel at /opt/conda/conda-bld/pytorch_1573049306803/work/aten/src/ATen/native/cuda/Loops.cuh:102) |
@lookthatdog i guess you ran your code on some device other than 0 right? And that GPU is of different type from GPU0? |
Eventually, this will help. |
@rusty1s yes, or in my case I specified the architectures for the |
Hello. my server has 4 GPUs. The default one is an RTX 2080. THe package can be installed and run fine on this GPU but when I run on other GPUs (TitanX), i got the following error
My guess is that
setup.py
automatically sets the flag for the RTX sm architecture. However, when I tried to addbut when installing i got this error
Please help!
The text was updated successfully, but these errors were encountered: