search

KailiWang1 / RLBind

a deep learning architecture for RNA-ligand binding sites prediction
Apache License 2.0
17 stars 3 forks source link

Unable to run code. #2

Open majumderS opened 10 months ago

majumderS commented 10 months ago

I have a gpu enabled system. I also have installed pytorch with cuda support:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia I am still unable to get the predict.py not running with the following message:

raise RuntimeError('Attempting to deserialize object on a CUDA ' RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

KailiWang1 commented 8 months ago

Please ensure that your environment supports GPU.