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
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.14andtensorflow-gpu=1.14and neither worked. Also I tried updating/changing the Google Colab connected GPU device to no avail. Any ideas?