videogames.ai Blog About Hardware guide

How to select which gpu tensorflow uses

Ex: selects GPU 6

export CUDA_VISIBLE_DEVICES=6

Ex: select no GPU, which makes tensorflow fallback to CPU

export CUDA_VISIBLE_DEVICES=