diff --git a/pt24-cu121/attachments/!FIRST_RUN-compile-install.bat b/pt24-cu121/attachments/!FIRST_RUN-compile-install.bat index cb20758..f8b9087 100644 --- a/pt24-cu121/attachments/!FIRST_RUN-compile-install.bat +++ b/pt24-cu121/attachments/!FIRST_RUN-compile-install.bat @@ -1,5 +1,5 @@ @REM Edit this to your GPU arch. -@REM The "+PTX" here is a fail-safe (forward compatibility), you don't need to add that. +@REM You don't need to add the "+PTX". Here it works as a fail-safe (providing forward compatibility). set TORCH_CUDA_ARCH_LIST=6.1+PTX @REM Compile-install PyTorch3D @@ -10,9 +10,19 @@ set CMAKE_ARGS=-DBUILD_opencv_world=ON -DWITH_CUDA=ON -DCUDA_FAST_MATH=ON -DWITH set PATH=%PATH%;%~dp0\python_embeded\Scripts .\python_embeded\python.exe -s -m pip install --force-reinstall ^ - "git+https://github.com/facebookresearch/pytorch3d.git@stable" + "git+https://github.com/facebookresearch/pytorch3d.git@V0.7.8" -@REM Compile-install pointnet2_ops for Triplane Gaussian Transformers +@REM Compile-install pointnet2_ops for Triplane Gaussian .\python_embeded\python.exe -s -m pip install --force-reinstall ^ .\extras\pointnet2_ops + +@REM Compile-install diff-gaussian-rasterization for Triplane Gaussian + +.\python_embeded\python.exe -s -m pip install --force-reinstall ^ + "git+https://github.com/ashawkey/diff-gaussian-rasterization.git" + +@REM Compile-install simple-knn + +.\python_embeded\python.exe -s -m pip install --force-reinstall ^ + .\extras\simple-knn diff --git "a/pt24-cu121/attachments/\344\270\255\346\226\207\350\204\232\346\234\254/!\351\246\226\346\254\241\350\277\220\350\241\214-\347\274\226\350\257\221\345\256\211\350\243\205\344\276\235\350\265\226\351\241\271.bat" "b/pt24-cu121/attachments/\344\270\255\346\226\207\350\204\232\346\234\254/!\351\246\226\346\254\241\350\277\220\350\241\214-\347\274\226\350\257\221\345\256\211\350\243\205\344\276\235\350\265\226\351\241\271.bat" index b7f3119..4e4bc89 100644 --- "a/pt24-cu121/attachments/\344\270\255\346\226\207\350\204\232\346\234\254/!\351\246\226\346\254\241\350\277\220\350\241\214-\347\274\226\350\257\221\345\256\211\350\243\205\344\276\235\350\265\226\351\241\271.bat" +++ "b/pt24-cu121/attachments/\344\270\255\346\226\207\350\204\232\346\234\254/!\351\246\226\346\254\241\350\277\220\350\241\214-\347\274\226\350\257\221\345\256\211\350\243\205\344\276\235\350\265\226\351\241\271.bat" @@ -11,9 +11,19 @@ set PATH=%PATH%;%~dp0\python_embeded\Scripts set PIP_INDEX_URL=https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple .\python_embeded\python.exe -s -m pip install --force-reinstall ^ - "git+https://ghp.ci/https://github.com/facebookresearch/pytorch3d.git@stable" + "git+https://ghp.ci/https://github.com/facebookresearch/pytorch3d.git@V0.7.8" -@REM 编译安装 pointnet2_ops ,该组件用于 Triplane Gaussian Transformers +@REM 编译安装 pointnet2_ops ,该组件用于 Triplane Gaussian .\python_embeded\python.exe -s -m pip install --force-reinstall ^ .\extras\pointnet2_ops + +@REM 编译安装 diff-gaussian-rasterization ,该组件用于 Triplane Gaussian + +.\python_embeded\python.exe -s -m pip install --force-reinstall ^ + "git+https://ghp.ci/https://github.com/ashawkey/diff-gaussian-rasterization.git" + +@REM 编译安装 simple-knn + +.\python_embeded\python.exe -s -m pip install --force-reinstall ^ + .\extras\simple-knn