Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
@@ -45,15 +45,14 @@ ENV HOME=/home/user \
|
|
45 |
GRADIO_THEME=huggingface \
|
46 |
SYSTEM=spaces
|
47 |
|
48 |
-
RUN pip install --upgrade pip
|
|
|
49 |
RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
|
50 |
|
51 |
RUN python -c "import torch; print(torch.version.cuda)"
|
52 |
COPY requirements.txt /tmp
|
53 |
RUN cd /tmp && pip install -r requirements.txt
|
54 |
|
55 |
-
RUN pip install setuptools==69.5.1
|
56 |
-
|
57 |
# install pointnet2_ops from snowflake
|
58 |
RUN git clone https://github.com/AllenXiangX/SnowflakeNet.git /home/user/SnowflakeNet
|
59 |
WORKDIR /home/user/SnowflakeNet/models/pointnet2_ops_lib
|
@@ -64,6 +63,7 @@ RUN git clone -b v0.7.3 https://github.com/facebookresearch/pytorch3d.git /home/
|
|
64 |
WORKDIR /home/user/pytorch3d-0.7.3
|
65 |
RUN python setup.py install --user
|
66 |
|
|
|
67 |
# install torch-scatter
|
68 |
RUN git clone https://github.com/rusty1s/pytorch_scatter.git /home/user/pytorch_scatter
|
69 |
WORKDIR /home/user/pytorch_scatter
|
|
|
45 |
GRADIO_THEME=huggingface \
|
46 |
SYSTEM=spaces
|
47 |
|
48 |
+
RUN pip install --upgrade pip ninja
|
49 |
+
RUN pip install setuptools==69.5.1
|
50 |
RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
|
51 |
|
52 |
RUN python -c "import torch; print(torch.version.cuda)"
|
53 |
COPY requirements.txt /tmp
|
54 |
RUN cd /tmp && pip install -r requirements.txt
|
55 |
|
|
|
|
|
56 |
# install pointnet2_ops from snowflake
|
57 |
RUN git clone https://github.com/AllenXiangX/SnowflakeNet.git /home/user/SnowflakeNet
|
58 |
WORKDIR /home/user/SnowflakeNet/models/pointnet2_ops_lib
|
|
|
63 |
WORKDIR /home/user/pytorch3d-0.7.3
|
64 |
RUN python setup.py install --user
|
65 |
|
66 |
+
RUN nvcc -V
|
67 |
# install torch-scatter
|
68 |
RUN git clone https://github.com/rusty1s/pytorch_scatter.git /home/user/pytorch_scatter
|
69 |
WORKDIR /home/user/pytorch_scatter
|