File size: 405 Bytes
c2e497e 0216e5a c2e497e |
1 2 3 4 5 6 7 8 9 10 11 |
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
|