Skip to content

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

Description

@binkiebinkie

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

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions