File size: 1,567 Bytes
87e9895 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | #!/bin/bash
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"
|