search

DeepRank / deeprank2

An open-source deep learning framework for data mining of protein-protein interfaces or single-residue variants.
https://deeprank2.readthedocs.io/en/latest/?badge=latest
Apache License 2.0
35 stars 10 forks source link

Fix usage for multiple GPUs #325

Open gcroci2 opened 1 year ago

gcroci2 commented 1 year ago

Everything works with 1 GPU and num_workers > 0, but if the number of GPUs is set to > 1 I get this error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0! (when checking argument for argument mat1 in method wrapper_addmm)

See these guides as starting point:

github-actions[bot] commented 1 year ago

This issue is stale because it has been open for 30 days with no activity. Remove stale label or comment or this will be closed in 7 days.