Skip to content

Working on gpu #20

Description

@Sean0417

Hi,
thank you very much for your work.
I followed your installation process, and install the exact packages in you requirements.txt. However there's always conflict using thos packages, could you just provide a more specific version on slurm? And right now I'm running into issues, it seems I can't use gpu for training? Seems like there's a conflict between xformers and pytorch?
(detany3d) v100dd25@a0125:/home/vault/v100dd/v100dd25/DetAny3D$ sh /home/vault/v100dd/v100dd25/DetAny3D/inference.sh
FutureWarning: Using TRANSFORMERS_CACHE is deprecated and will be removed in v5 of Transformers. Use HF_HOME instead.
FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers
UserWarning: Failed to load custom C++ ops. Running on CPU mode Only!
xFormers not available
Start running basic DDP example on rank 0.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions