Spaces:
Running
on
T4
Running
on
T4
Jitesh Jain
commited on
Commit
·
c0d4288
1
Parent(s):
3f01ea0
Fix user
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -41,7 +41,7 @@ ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
|
|
| 41 |
RUN pip install ninja
|
| 42 |
|
| 43 |
USER root
|
| 44 |
-
RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1
|
| 45 |
|
| 46 |
RUN sh deform_setup.sh
|
| 47 |
|
|
|
|
| 41 |
RUN pip install ninja
|
| 42 |
|
| 43 |
USER root
|
| 44 |
+
RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build install --user && cd ..
|
| 45 |
|
| 46 |
RUN sh deform_setup.sh
|
| 47 |
|