Skip to content

Latest commit

 

History

History
17 lines (10 loc) · 966 Bytes

File metadata and controls

17 lines (10 loc) · 966 Bytes

CI Codespaces Prebuilds

Sentiment Analysis on IMDb data with Naive Bayes and BERT Models

Verify GPU works

The following examples test out the GPU

  • run pytorch training test: python utils/quickstart_pytorch.py
  • run pytorch CUDA test: python utils/verify_cuda_pytorch.py
  • run tensorflow training test: python utils/quickstart_tf2.py
  • run nvidia monitoring test: nvidia-smi -l 1 it should show a GPU
  • run whisper transcribe test ./utils/transcribe-whisper.sh and verify GPU is working with nvidia-smi -l 1