search

sandialabs / cross-sim

CrossSim: accuracy simulation of analog in-memory computing
Other
117 stars 26 forks source link

If GPU acceleration is unabled #11

Closed longcheng2022 closed 8 months ago

longcheng2022 commented 1 year ago

My computer's GPU acceleration is unabled, could you teach me how to operate the MLP_training.py? Do I need to remove all GPU-related code? (I have a try to delete some code, but once I run it, it reports an error "not implemented error"). Thanks for your kind help.

ptxiao commented 1 year ago

Have you tried setting useGPU = False? The code should work the same way if you don't have a compatible GPU, you shouldn't need to make any other changes to the code.