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.

Training.py not running on GPU

See original GitHub issue

Hi @f90 , I have a GPU in my paperspace machine. I am trying to run Training.py using the following command: python Training.py with cfg.full_44KHz. But the script runs only on CPU, not on GPU. I have just forked this repo and executing the Training.py script. How can I make sure that it runs on GPU. I am a Pytorch guy, and trying to refactor the codebase for Pytorch. Please help. Thak you

Issue Analytics

State:
Created 4 years ago
Comments:6 (3 by maintainers)

Top GitHub Comments

1reaction

imflash217commented, Jul 29, 2019

Updating the Nvidia driver to version-396+ solved the issue. Thanks @f90

1reaction

f90commented, Jul 25, 2019

This seems like an NVIDIA issue, not related to my code.

Whats your NVIDIA Driver, CUDA and CuDNN versions? Looks like cuDNN version might not be the right one. Tensorflow 1.8 requires CUDA 9, and I have installed cuDNN 7.5.0 alongside and it works. I also noticed your driver seems quite old according to the logs, maybe you can update that?

Top Results From Across the Web

Training process doesn't run on GPU #1739 - GitHub

Hi Mr, when i run your traning source code with only person class. I already set GPU_COUNT to 2 but it did not...

Training don't start with GPU, but running fine with CPU ...

I am using Titan V for training. I think there is a Problem with the command I am using. If I use CPU...

Tensorflow not running on GPU - Stack Overflow

I think when I uninstalled tensorflow it deleted the init.py file or something. So I ran pip install --ignore-installed --upgrade and now this ......

Running Python script on GPU. - GeeksforGeeks

First, make sure that Nvidia drivers are upto date also you can install cudatoolkit explicitly from here. then install Anaconda add anaconda to ......

Can't make Tensorflow run on GPU - Ask Ubuntu

Enter the Python shell (something like python3 in your terminal) >>> import tensorflow as tf >>> tf.config.list_physical_devices("GPU") ...