Ask Your Question

Why does Pytorch training work on CPU but not GPU?

asked 2022-10-15 11:00:00 +0000

pufferfish gravatar image

edit retag flag offensive close merge delete

1 Answer

Sort by ยป oldest newest most voted

answered 2021-06-15 15:00:00 +0000

huitzilopochtli gravatar image

It's highly unlikely that Pytorch training works on CPU but not GPU. GPUs are generally used for training large and complex models in deep learning frameworks like Pytorch because of their high parallel processing capabilities that significantly speed up the training process. It's possible that the error is due to a hardware or software issue with the GPU, such as malfunctioning hardware, incorrect driver installation, or incorrect CUDA installation. It's worth checking the hardware specifications, driver versions, and versions of Pytorch and CUDA to ensure compatibility. Additionally, some models and data may not require or benefit from using a GPU, which may lead to the misconception that Pytorch training does not work on GPUs.

edit flag offensive delete link more

Your Answer

Please start posting anonymously - your entry will be published after you log in or create a new account. This space is reserved only for answers. If you would like to engage in a discussion, please instead post a comment under the question or an answer that you would like to discuss

Add Answer

Question Tools


Asked: 2022-10-15 11:00:00 +0000

Seen: 2 times

Last updated: Jun 15 '21