Logo

Indie Machine Learning and Video Game Dev

Neural Net Downloads
How to
Hardware Reviews
Get Merch
About

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