lqg f7e659eb0b add ai 2 years ago
..
.devcontainer f7e659eb0b add ai 2 years ago
.github f7e659eb0b add ai 2 years ago
utils f7e659eb0b add ai 2 years ago
.gitignore f7e659eb0b add ai 2 years ago
Makefile f7e659eb0b add ai 2 years ago
README.md f7e659eb0b add ai 2 years ago
environment.yml f7e659eb0b add ai 2 years ago
requirements.txt f7e659eb0b add ai 2 years ago

README.md

ai

Repo that allows me to build AI tools on top of Hugging Face

How to do fast inference using API

Verify GPU working

  • Reference PyTorch site
  • numba -s | grep cuda
  • run python utils/verify_cuda_pytorch.py
  • run nvidia-smi should show a GPU