stylegan2-flax-tpu icon indicating copy to clipboard operation
stylegan2-flax-tpu copied to clipboard

Published 20 hours ago • verified publisher icon nyx-ai

TPU errors?? Anyone else correct this yet?

Open jimb2834 opened this issue 2 years ago • 2 comments

Hello,

I was trying to test and got this error. I am using Colab Pro+ and have a TPU/High Mem.

Any idea why this occurs?

2022-12-22 13:40:59.434952: E external/org_tensorflow/tensorflow/compiler/xla/stream_executor/cuda/cuda_driver.cc:267] failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected WARNING:jax._src.lib.xla_bridge:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)

jimb2834 avatar Dec 22 '22 13:12 jimb2834