Shinobi/plugins/tensorflow/.env

3 lines
57 B
Bash

TF_FORCE_GPU_ALLOW_GROWTH=true
#CUDA_VISIBLE_DEVICES=0,2