sinzlab / sensorium

Code base for the SENSORIUM competition.
https://sensorium2022.net/
MIT License
59 stars 32 forks source link

RuntimeError: No CUDA GPUs are available even when using device='cpu' #104

Open aaprasad opened 2 years ago

aaprasad commented 2 years ago

When running submission.generate_submission_file(trained_model=model, dataloaders=dataloaders, data_key=dataset_name, path="./submission_files/", device="cpu") I still get a RuntimeError: No CUDA GPUs are available even when using device='cpu'

KonstantinWilleke commented 2 years ago

Thanks, we'll look into that one!