Spaces:
Running
on
T4
Running
on
T4
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -27,7 +27,7 @@ RUN python3 -m spacy download en_core_web_sm
|
|
27 |
|
28 |
COPY pip.conf $VIRTUAL_ENV/pip.conf
|
29 |
|
30 |
-
RUN pip3 install onnxruntime-
|
31 |
|
32 |
RUN useradd -m -u 1000 user
|
33 |
USER user
|
|
|
27 |
|
28 |
COPY pip.conf $VIRTUAL_ENV/pip.conf
|
29 |
|
30 |
+
RUN pip3 install onnxruntime-gpu==1.17.1
|
31 |
|
32 |
RUN useradd -m -u 1000 user
|
33 |
USER user
|