You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
The text was updated successfully, but these errors were encountered:
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
andtensorflow-gpu=1.14
and neither worked. Also I tried updating/changing the Google Colab connected GPU device to no avail. Any ideas?The text was updated successfully, but these errors were encountered: