Update Dockerfile
Browse files- Dockerfile +8 -0
Dockerfile
CHANGED
|
@@ -1,2 +1,10 @@
|
|
| 1 |
FROM huggingface/autotrain-advanced:7f5ff64
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1
|
|
|
|
| 1 |
FROM huggingface/autotrain-advanced:7f5ff64
|
| 2 |
+
|
| 3 |
+
# Install CUDA toolkit and dependencies
|
| 4 |
+
RUN apt-get update && apt-get install -y \
|
| 5 |
+
cuda-toolkit-11-2 \
|
| 6 |
+
libcudnn8=8.1.1.33-1+cuda11.2 \
|
| 7 |
+
libcudnn8-dev=8.1.1.33-1+cuda11.2
|
| 8 |
+
|
| 9 |
+
# Uninstall and reinstall AutoTrain Advanced
|
| 10 |
CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1
|