Open hans opened 4 years ago
Could you please add some more instructions on how to enable GPU usage for the other models not mentioned in #49? We're trying to use some larger pre-trained models with syntaxgym and can't get them to use the available GPUs.
Use some new environment variable e.g.
LMZOO_GPU_ENABLED
.CUDA_VISIBLE_DEVICES
gets passed through on all backends