search

JSchlensok / VespaG

Expert-Guided Protein Language Models enable Accurate and Blazingly Fast Fitness Prediction
GNU General Public License v3.0
8 stars 3 forks source link

CUDA memory issues #10

Open JSchlensok opened 2 months ago

JSchlensok commented 2 months ago

Multiple users have reported CUDA running out of memory even with ~40GB of VRAM when loading embeddings into RAM.

I haven't yet been able to reproduce this error, will try with a fresh environment.