Skip to content

Instantly share code, notes, and snippets.

View wangrui6's full-sized avatar

Ray Wang wangrui6

View GitHub Profile
@j-min
j-min / test_multi_gpu.py
Last active January 15, 2022 01:57
TensorFlow multi GPU example
from __future__ import print_function
'''
Basic Multi GPU computation example using TensorFlow library.
Author: Aymeric Damien
Project: https://github.com/aymericdamien/TensorFlow-Examples/
'''
'''
This tutorial requires your machine to have 2 GPUs
"/cpu:0": The CPU of your machine.