Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Running on CPU without using GPU. In case there is no CUDA compatibility. #18

Open
parsapahlavan opened this issue Oct 11, 2023 · 1 comment

Comments

@parsapahlavan
Copy link

Is it possible to run this on a machine with no GPU? Also, is it possible to run it on a machine that is not CUDA-enabled?

@Mohammadakhavan75
Copy link

Mohammadakhavan75 commented Dec 9, 2023

I found a workaround for this.

You need to comment line 47, and edit line 137 in spiga/framework.py file.

line 47:
# self.model = self.model.cuda(gpus[0])
and change line 137:
# data_var = data.cuda(device=self.gpus[0], non_blocking=True)
into this:
data_var = data

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants