Skip to content

Commit

Permalink
clone 'pt21-cu118' to 'pt23-cu121'
Browse files Browse the repository at this point in the history
  • Loading branch information
YanWenKun committed Dec 11, 2024
1 parent 0c91035 commit 7b5ea13
Show file tree
Hide file tree
Showing 24 changed files with 8,077 additions and 0 deletions.
42 changes: 42 additions & 0 deletions pt23-cu121/attachments/!FIRST_RUN-compile-install.bat
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
@REM Edit this to your GPU arch.
@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
@REM PyTorch3D on Windows works best through compile-install. Binary-install will fail on some workflows.
@REM e.g. "CUDA error: no kernel image is available for execution on the device"

set CMAKE_ARGS=-DBUILD_opencv_world=ON -DWITH_CUDA=ON -DCUDA_FAST_MATH=ON -DWITH_CUBLAS=ON -DWITH_NVCUVID=ON
set PATH=%PATH%;%~dp0\python_embeded\Scripts

.\python_embeded\python.exe -s -m pip install ^
"git+https://github.com/facebookresearch/[email protected]"

@REM Compile-install pointnet2_ops for Triplane Gaussian

.\python_embeded\python.exe -s -m pip install ^
.\extras\pointnet2_ops

@REM Compile-install diff-gaussian-rasterization for Triplane Gaussian

.\python_embeded\python.exe -s -m pip install ^
"git+https://github.com/ashawkey/diff-gaussian-rasterization.git"

@REM Compile-install simple-knn

.\python_embeded\python.exe -s -m pip install ^
.\extras\simple-knn

@REM Compile-install kiuikit

.\python_embeded\python.exe -s -m pip install ^
git+https://github.com/ashawkey/kiuikit.git

@REM Compile-install nvdiffrast

.\python_embeded\python.exe -s -m pip install ^
git+https://github.com/NVlabs/nvdiffrast.git

@REM Ensure numpy1

.\python_embeded\python.exe -s -m pip install numpy==1.26.4
Loading

0 comments on commit 7b5ea13

Please sign in to comment.