Skip to content

Working on gpu #20

@Sean0417

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