Skip to content

torch verision conflict with the arch with A100/3090? #5

Description

@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

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions