/test-gpu-tensorflow

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

Primary LanguagePython

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