forcing pytorch3d ver

YanWenKun · Dec 8, 2024 · 886e62e · 886e62e
1 parent 649ce27
commit 886e62e
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 5 deletions.
diff --git 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/中文脚本/!首次运行-编译安装依赖项.bat b/pt24-cu121/attachments/中文脚本/!首次运行-编译安装依赖项.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