Spaces:
Running
on
T4
Running
on
T4
Commit
•
2685271
1
Parent(s):
be0b089
Fix user
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -35,11 +35,11 @@ RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
|
|
| 35 |
|
| 36 |
COPY . .
|
| 37 |
|
| 38 |
-
RUN sh deform_setup.sh
|
| 39 |
-
|
| 40 |
ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
|
| 41 |
|
| 42 |
-
RUN ln -s ./oneformer/modeling/pixel_decoder/ops/ ./ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR install --user && cd ..
|
|
|
|
|
|
|
| 43 |
|
| 44 |
# USER user
|
| 45 |
|
|
|
|
| 35 |
|
| 36 |
COPY . .
|
| 37 |
|
|
|
|
|
|
|
| 38 |
ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
|
| 39 |
|
| 40 |
+
RUN ln -s ./oneformer/modeling/pixel_decoder/ops/ ./ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/oneformer/ install --user && cd ..
|
| 41 |
+
|
| 42 |
+
RUN sh deform_setup.sh
|
| 43 |
|
| 44 |
# USER user
|
| 45 |
|