e82f02a9a5
Release 3D scene generation pipeline and tag as v0.1.2. --------- Co-authored-by: xinjie.wang <xinjie.wang@gpu-4090-dev015.hogpu.cc>
53 lines
1.6 KiB
Bash
53 lines
1.6 KiB
Bash
#!/bin/bash
|
|
set -e
|
|
SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
source "$SCRIPT_DIR/_utils.sh"
|
|
|
|
# try_install "Installing txt2panoimg..." \
|
|
# "pip install txt2panoimg@git+https://github.com/HochCC/SD-T2I-360PanoImage --no-deps" \
|
|
# "txt2panoimg installation failed."
|
|
|
|
# try_install "Installing fused-ssim..." \
|
|
# "pip install fused-ssim@git+https://github.com/rahul-goel/fused-ssim#egg=328dc98" \
|
|
# "fused-ssim installation failed."
|
|
|
|
# try_install "Installing tiny-cuda-nn..." \
|
|
# "pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch" \
|
|
# "tiny-cuda-nn installation failed."
|
|
|
|
# try_install "Installing pytorch3d" \
|
|
# "pip install git+https://github.com/facebookresearch/pytorch3d.git@v0.7.7" \
|
|
# "pytorch3d installation failed."
|
|
|
|
|
|
PYTHON_PACKAGES_NODEPS=(
|
|
timm
|
|
txt2panoimg@git+https://github.com/HochCC/SD-T2I-360PanoImage
|
|
kornia
|
|
kornia_rs
|
|
)
|
|
|
|
PYTHON_PACKAGES=(
|
|
fused-ssim@git+https://github.com/rahul-goel/fused-ssim#egg=328dc98
|
|
git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
|
|
git+https://github.com/facebookresearch/pytorch3d.git@v0.7.7
|
|
h5py
|
|
albumentations==0.5.2
|
|
webdataset
|
|
icecream
|
|
open3d
|
|
pyequilib
|
|
numpy==1.26.4
|
|
triton==2.1.0
|
|
)
|
|
|
|
for pkg in "${PYTHON_PACKAGES_NODEPS[@]}"; do
|
|
try_install "Installing $pkg without dependencies..." \
|
|
"pip install --no-deps $pkg" \
|
|
"$pkg installation failed."
|
|
done
|
|
|
|
try_install "Installing other Python dependencies..." \
|
|
"pip install ${PYTHON_PACKAGES[*]}" \
|
|
"Python dependencies installation failed."
|