spam classifier using tflite-model-maker
75
This image correctly setup python and all necessary packages require to run tflite-model-maker packages with nvidia gpu accelerated capability. Model building and inference can be done in this image.
With the following to check if the gpu can be shown correctly in the docker container
docker run --gpus all -it -v "$(pwd)":/workspace -w /workspace jasonwee/spam-model:gpu-nvidia bash
python -c "import tensorflow as tf; print('Num GPUs Available: ', len(tf.config.list_physical_devices('GPU')))"
docker run --gpus all -it -v "$(pwd)":/workspace -w /workspace jasonwee/spam-model:tf2.21-gpu python -c "import tensorflow as tf; print('Num GPUs Available: ', len(tf.config.list_physical_devices('GPU')))"
Content type
Image
Digest
sha256:08b9445c8…
Size
3.4 GB
Last updated
6 days ago
docker pull jasonwee/spam-model:tf2.21-gpu