Logo

Indie Machine Learning and Video Game Dev

Neural Net Downloads
How to
Hardware Reviews
Get Merch
About
Training of the week

How to select which gpu tensorflow uses

by Mathieu Poliquin

Ex: selects GPU 6

export CUDA_VISIBLE_DEVICES=6

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

export CUDA_VISIBLE_DEVICES=
tags: tensorflow - gpu