Skip to content

[Request for Error Troubleshooting] RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc) #5

@kskim-phd

Description

@kskim-phd

Dear Team;

Thank you for sharing this great platform!

I have set up according to your instruction but I got the following error when I command the training code :

python train_net.py
--config-file configs/esm35_evo1_initial_pdbonly.yaml
--num-gpus 1 SOLVER.SEQ_PER_BATCH 1
OUTPUT_DIR output/esm35_evo1/initial_pdbonly

This is error : RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

Screenshot from 2024-02-25 12-00-09

I have run this code with 1 GPU A6000.

Could you reply on how to solve this issue if possible? I deeply appreciate it.

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