87ff24dbd4
Update image-to-3d model loading position. --------- Co-authored-by: xinjie.wang <xinjie.wang@gpu-4090-dev015.hogpu.cc>
36 lines
926 B
Bash
36 lines
926 B
Bash
#!/bin/bash
|
|
set -e
|
|
SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
source "$SCRIPT_DIR/_utils.sh"
|
|
|
|
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."
|