| FROM huggingface/autotrain-advanced:7f5ff64 | |
| # Install CUDA toolkit and dependencies | |
| RUN apt-get update && apt-get install -y \ | |
| cuda-toolkit-11-2 \ | |
| libcudnn8=8.1.1.33-1+cuda11.2 \ | |
| libcudnn8-dev=8.1.1.33-1+cuda11.2 | |
| # Uninstall and reinstall AutoTrain Advanced | |
| CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1 | |