search

SHI-Labs / Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
MIT License
1.05k stars 86 forks source link

Question on the compilation with different version of cuda or pytorch #5

Closed PeiqinZhuang closed 2 years ago

PeiqinZhuang commented 2 years ago

Thanks for this work. I just wonder if the Cuda source code could be compiled with different versions of PyTorch or Cuda driver which you recommend? Right now, I have some problems with older pytorch and cuda driver for the compilation.

alihassanijr commented 2 years ago

Hi, thank you for your interest. It should be compatible with recent PyTorch and CUDA versions. We've only tried torch 1.8 and above, and strongly recommend the latest, torch 1.11. Could you share the error you're facing and the torch/cuda versions you have?

PeiqinZhuang commented 2 years ago

Hi, thank you for your interest. It should be compatible with recent PyTorch and CUDA versions. We've only tried torch 1.8 and above, and strongly recommend the latest, torch 1.11. Could you share the error you're facing and the torch/cuda versions you have?

Hi, Thanks for the reply. I attach the screenshot with the error. Actually, I use torch 1.8 and Cuda driver 10.1 截屏2022-04-19 上午10 10 17

stevenwalton commented 2 years ago

Hi Peiqin, what GPU are you using? I don't think we have tested that cuda driver but are looking into it.

alihassanijr commented 2 years ago

This is really strange. Seems to be more of a torch error than anything else. Would you mind also sharing the entire error and the command you ran, and as many details as possible on the environment, so we try and see if we can reproduce it on our end?

PeiqinZhuang commented 2 years ago

Hi Peiqin, what GPU are you using? I don't think we have tested that cuda driver but are looking into it.

Hi, I just compiled it on V100.

alihassanijr commented 2 years ago

Hi, Could you share the command you ran and your environment details so we can look into the issue?

PeiqinZhuang commented 2 years ago

Hi, Could you share the command you ran and your environment details so we can look into the issue?

Hi, I just solved this problem by creating a new python environment. It seems that there exist some problems with my previous environment.

alihassanijr commented 2 years ago

Thank you for your response. If you don't have any other questions, feel free to close this issue. Otherwise feel free to ask.

alihassanijr commented 2 years ago

Closing this due to inactivity. If you still have questions feel free to open it back up.