GPU to use

[snapo]

Is there a way to tell tensorneat to use GPU:1 instead of GPU:0 ?

[WLS2002]

Maybe you can try these two methods:
1.
Use export CUDA_VISIBLE_DEVICES=1 before running python script.
or
Add these codes at the begining of your code:

import os
os.environ['CUDA_VISIBLE_DEVICES']='1'

Add these codes at the begining:

import jax
jax.config.update("jax_default_device", jax.devices("gpu")[1])

I hope this can help you, and thank you for using TensorNEAT!

[snapo]

thank you will give it a try as soon as the current run finnishes (approx. 30-40h)