GPU instead of CPU - Githubissues

mattwang44 / LeNet-from-Scratch

Implementation of LeNet5 without any auto-differentiate tools or deep learning frameworks. Accuracy of 98.6% is achieved on MNIST dataset.

26 stars 9 forks source link

GPU instead of CPU #2

Open JEGarcia10 opened 2 years ago

JEGarcia10 commented 2 years ago

Has there been any updates to running the code using GPU, instead of CPU? I am new to python and running a GEFORCE RTX 3080, but of course as stated in the README, it is using CPU, which takes longer then running with GPU.