zouzx commited on
Commit
23bebfb
1 Parent(s): d58eec5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 setuptools ninja
 
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