Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Using GPU to train the model

See original GitHub issue

Hello, I’m really appreciate your work. But now I wonder how to use GPU to train the model. There are always mistakes when I use the CUDA device. Thanks a lot. device = torch.device('cuda' if use_cuda else 'cpu')

Issue Analytics

State:
Created 3 years ago
Comments:7 (3 by maintainers)

Top GitHub Comments

1reaction

penny9287commented, Jan 14, 2021

Thanks a lot. I removed the above line and the GPU works!

0reactions

realsarmcommented, May 30, 2022

@jettify #439 This is a fix for shampoo optimizer by upgrading its implementation

Top Results From Across the Web

How to train Tensorflow models. Using GPUs | by DeviceHive

So, how would one approach using GPUs for machine learning tasks? In this post we will explore the setup of a GPU-enabled AWS...

Using GPUs for training models in the cloud | AI Platform ...

Graphics Processing Units (GPUs) can significantly accelerate the training process for many deep learning models. Training models for tasks like image ...

Use a GPU | TensorFlow Core

Developing for multiple GPUs will allow a model to scale with the additional resources. If developing on a system with a single GPU,...

Machine Learning on GPU - GitHub Pages

In this lesson we will consider different ways of measuring performance and draw comparisons between training a model on the CPU and training...

Deep Learning GPU: Making the Most of GPUs for Your Project

GPUs are commonly used for deep learning, to accelerate training and inference for computationally intensive models. Keras is a Python-based, deep learning API ......