torch verision conflict with the arch with A100/3090?

[cheliu-computation]

Hi, thanks for your impressive work.

I just got some issue when I attemped to deploy this repo on my cluster (with 8 A100 GPU)
First I request more than 100Gb memory but still got the segmentation fault error, it is about the memory leaking, does it mean there is any memory sensitive operation in the code?

Also, when I check the package verision, I found the torch is set as 1.9, but my torch is 1.13 with CUDA 11.6.
Does this code only work with the torch 1.9, becasue the current arch GPU, such as A100/A6000/3090 only support torch>1.11?

Looking forward to your resposne!
Best Regards

[chao1224]

Hi @cheliu-computation,

Can you provide more details on what CMD you are running now?
BTW. Now our Python scripts are for single-gpu running, and no DDP has been added yet.