| source /home/hubin/venvs/trellis/bin/activate | |
| export TORCH_CUDA_ARCH_LIST="9.0" | |
| export MAX_JOBS=32 | |
| export FORCE_CUDA=1 | |
| EXT=/home/hubin/trellis_work/extensions | |
| LOG=/home/hubin/trellis_work/build_logs | |
| mkdir -p $EXT $LOG | |
| cd /home/hubin/trellis_work | |
| step() { | |
| name=$1; shift | |
| echo "[$(date +%T)] START $name" | |
| if "$@" > $LOG/$name.log 2>&1; then echo "[$(date +%T)] OK $name"; else echo "[$(date +%T)] FAIL $name (see $LOG/$name.log)"; fi | |
| } | |
| step basic pip install imageio imageio-ffmpeg tqdm easydict opencv-python-headless ninja trimesh tensorboard pandas lpips zstandard kornia "timm>=1.0" \ | |
| "git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8" | |
| clone() { [ -d $EXT/$1 ] || git clone $3 $2 $EXT/$1; } | |
| step clone_nvdiffrast clone nvdiffrast https://github.com/NVlabs/nvdiffrast.git "-b v0.4.0" | |
| step clone_nvdiffrec clone nvdiffrec https://github.com/JeffreyXiang/nvdiffrec.git "-b renderutils" | |
| step clone_cumesh clone CuMesh https://github.com/JeffreyXiang/CuMesh.git "--recursive" | |
| step clone_flexgemm clone FlexGEMM https://github.com/JeffreyXiang/FlexGEMM.git "--recursive" | |
| step nvdiffrast pip install $EXT/nvdiffrast --no-build-isolation | |
| step nvdiffrec pip install $EXT/nvdiffrec --no-build-isolation | |
| step cumesh pip install $EXT/CuMesh --no-build-isolation | |
| step flexgemm pip install $EXT/FlexGEMM --no-build-isolation | |
| rm -rf $EXT/o-voxel && cp -r TRELLIS.2/o-voxel $EXT/o-voxel | |
| step o_voxel pip install $EXT/o-voxel --no-build-isolation | |
| echo "[$(date +%T)] ALL DONE" | |