change conda to pip

README.md

@@ -24,6 +24,11 @@ Please refer to `install.sh` for manual installation. This script has been teste
 
 ### Primitive extraction
 
+### Evaluation
+
+### Baselines
+
+Please refer to README.md in the `baselines` folder
 
 ## Training
 

docker/Dockerfile_with_proxy

@@ -82,7 +82,7 @@ RUN echo 'channels:' > /root/.condarc && \
     echo '  deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/' >> /root/.condarc; \
     conda clean -i -y
 
-RUN git pull && chmod +x install_python.sh && ./install_python.sh
+RUN cd /root/NVDNet/ && git reset --hard && git pull && chmod +x install_python.sh && ./install_python.sh
 
 EXPOSE 22 6100 6101 6102 15000 15001
 CMD ["/usr/sbin/sshd", "-D"]

install_python.sh

@@ -6,7 +6,10 @@ ROOT_DIR=$PWD
 
 conda config --set auto_update_conda false
 # conda install -c intel mkl mkl-devel mkl-static mkl-include -y
-conda install -c pytorch -c nvidia pytorch=2.0.1 torchvision=0.15.2 torchaudio=2.0.2 pytorch-cuda=11.8 -y
-conda install -c pyg pyg  -y
+# conda install -c pytorch -c nvidia pytorch=2.0.1 torchvision=0.15.2 torchaudio=2.0.2 pytorch-cuda=11.8 -y
+pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
+# conda install -c pyg pyg  -y
+pip install torch_geometric 
+pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.0+cu118.html
 pip install PyMCubes pytorch-lightning hydra-core shapely scikit-image matplotlib tensorboard plyfile opencv-python opencv-contrib-python open3d ray[default] h5py
 pip install torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.1+cu118.html