Hello. I’d like to share my FastAPI-Fastai2 starter template. I had some problems getting my model, trained in colab, to work with my FastAPI code (on windows). And hopefully this can alleviate some headaches for others.
P.S. This wonderful starlette template was my reference. And I want to thank @muellerzr for it. And thanks to @thomascountz for his help with dockerfiles and version issues.
Thanks a lot for this!
I tried to run it on my machine but I’m getting
/usr/local/lib/python3.8/dist-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:100.)
Is there a way this can run on a CPU machine? or this has to be a GPU?