|
source /hpc2ssd/softwares/anaconda3/bin/activate instantmesh |
|
module load cuda/12.1 compilers/gcc-11.1.0 compilers/icc-2023.1.0 cmake/3.27.0 |
|
export CXX=$(which g++) |
|
export CC=$(which gcc) |
|
export CPLUS_INCLUDE_PATH=/hpc2ssd/softwares/cuda/cuda-12.1/targets/x86_64-linux/include:$CPLUS_INCLUDE_PATH |
|
export CUDA_LAUNCH_BLOCKING=1 |
|
export NCCL_TIMEOUT=3600 |
|
export CUDA_VISIBLE_DEVICES="0" |
|
|
|
python run.py configs/PRM_inference.yaml examples/恐龙套装.webp \ |
|
--seed 10 \ |
|
--materials "(0.0, 0.9)" \ |
|
--env_path "./env_mipmap/6" \ |
|
--output_path "output/" \ |
|
--save_video \ |
|
--export_texmap \ |
|
|
