hu-po / test-gpu-tensorflow

Mock Weights & Biases training project that uses a NVIDIA GPU.

Test Repo

This repo runs a mock Weights & Biases training project that uses a NVIDIA GPU.

Dependency	Version
Tensorflow	2.5.0
CUDA	11.0

The entry point for this repo is train.py