Logo

Indie Machine Learning and Video Game Dev

Neural Net Downloads
How to
Useful links
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