Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

When running training, RuntimeError: No GPU devices found error #32

Open
binkiebinkie opened this issue Oct 12, 2022 · 0 comments
Open

Comments

@binkiebinkie
Copy link

Hello, I can't get past the training step when training on Google Colab. I receive the error below. I have tried installing tensorflow=1.14 and tensorflow-gpu=1.14 and neither worked. Also I tried updating/changing the Google Colab connected GPU device to no avail. Any ideas?

Skipping registering GPU devices...
Failed!
Traceback (most recent call last):
  File "train.py", line 645, in <module>
    main()
  File "train.py", line 637, in main
    run_training(**vars(args))
  File "train.py", line 522, in run_training
    training_loop.training_loop(**training_options)
...
File "/content/drive/MyDrive/colab-sg2-ada/stylegan2-ada/dnnlib/tflib/ops/fused_bias_act.py", line 18, in _get_plugin
    return custom_ops.get_plugin(os.path.splitext(__file__)[0] + '.cu')
  File "/content/drive/MyDrive/colab-sg2-ada/stylegan2-ada/dnnlib/tflib/custom_ops.py", line 139, in get_plugin
    compile_opts += f' --gpu-architecture={_get_cuda_gpu_arch_string()}'
  File "/content/drive/MyDrive/colab-sg2-ada/stylegan2-ada/dnnlib/tflib/custom_ops.py", line 60, in _get_cuda_gpu_arch_string
    raise RuntimeError('No GPU devices found')
RuntimeError: No GPU devices found
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant