Spaces:
Running
on
T4
Running
on
T4
Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -22,6 +22,7 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
|
|
| 22 |
WORKDIR /data
|
| 23 |
|
| 24 |
ENV VIRTUAL_ENV=/opt/venv
|
|
|
|
| 25 |
|
| 26 |
RUN python3 -m venv $VIRTUAL_ENV
|
| 27 |
ENV PATH="$VIRTUAL_ENV/bin:$PATH"
|
|
|
|
| 22 |
WORKDIR /data
|
| 23 |
|
| 24 |
ENV VIRTUAL_ENV=/opt/venv
|
| 25 |
+
TORCH_CUDA_ARCH_LIST "7.5"
|
| 26 |
|
| 27 |
RUN python3 -m venv $VIRTUAL_ENV
|
| 28 |
ENV PATH="$VIRTUAL_ENV/bin:$PATH"
|