1

2026-04-13 11:20:56 +08:00
parent 0025b997a2
commit 95166ef698
43 changed files with 21 additions and 0 deletions
--- a/.gitignore
+++ b/.gitignore
--- a/.gitmodules
+++ b/.gitmodules
--- a/.idea/.gitignore
+++ b/.idea/.gitignore
--- a/0_glb_to_obj.py
+++ b/0_glb_to_obj.py
--- a/0_remesh.py
+++ b/0_remesh.py
--- a/1_obj_to_step.py
+++ b/1_obj_to_step.py
--- a/2_step_to_svg.py
+++ b/2_step_to_svg.py
--- a/21
+++ b/21
@@ -0,0 +1,21 @@
+由于服务器架构不同,需要重新构建pytorch3D 和 nvdiffrast
+
+conda activate trellis
+pip uninstall -y pytorch3d nvdiffrast
+export TORCH_CUDA_ARCH_LIST="8.6;8.9;9.0;12.0"
+pip install setuptools wheel ninja --upgrade
+pip install git+https://github.com/facebookresearch/pytorch3d.git --no-build-isolation --no-cache-dir
+pip install git+https://github.com/NVlabs/nvdiffrast.git --no-build-isolation --no-cache-dir
+
+python -c "
+import torch
+import pytorch3d
+import nvdiffrast.torch as dr
+print('PyTorch:', torch.__version__)
+print('GPU:', torch.cuda.get_device_name(0))
+print('Capability:', torch.cuda.get_device_capability(0))
+print('pytorch3d version:', pytorch3d.__version__)
+ctx = dr.RasterizeCudaContext()
+print('✅ nvdiffrast CUDA context OK')
+print('✅ pytorch3d should now work on A6000')
+"
--- a/CODE_OF_CONDUCT.md
+++ b/CODE_OF_CONDUCT.md
--- a/DATASET.md
+++ b/DATASET.md
--- a/app.py
+++ b/app.py
--- a/app_text.py
+++ b/app_text.py
--- a/blender_glb_to_obj.py
+++ b/blender_glb_to_obj.py
--- a/client.py
+++ b/client.py
--- a/dataset_toolkits/build_metadata.py
+++ b/dataset_toolkits/build_metadata.py
--- a/dataset_toolkits/calculate_aesthetic_scores.py
+++ b/dataset_toolkits/calculate_aesthetic_scores.py
--- a/dataset_toolkits/datasets/3D-FUTURE.py
+++ b/dataset_toolkits/datasets/3D-FUTURE.py
--- a/dataset_toolkits/datasets/ABO.py
+++ b/dataset_toolkits/datasets/ABO.py
--- a/trellis/datasets/init.py
+++ b/trellis/datasets/init.py
--- a/trellis/datasets/components.py
+++ b/trellis/datasets/components.py
--- a/trellis/models/init.py
+++ b/trellis/models/init.py
--- a/trellis/models/structured_latent_vae/init.py
+++ b/trellis/models/structured_latent_vae/init.py
--- a/trellis/models/structured_latent_vae/base.py
+++ b/trellis/models/structured_latent_vae/base.py
--- a/trellis/models/structured_latent_vae/decoder_gs.py
+++ b/trellis/models/structured_latent_vae/decoder_gs.py
--- a/trellis/models/structured_latent_vae/decoder_mesh.py
+++ b/trellis/models/structured_latent_vae/decoder_mesh.py
--- a/trellis/models/structured_latent_vae/decoder_rf.py
+++ b/trellis/models/structured_latent_vae/decoder_rf.py
--- a/trellis/modules/sparse/transformer/init.py
+++ b/trellis/modules/sparse/transformer/init.py
--- a/trellis/modules/sparse/transformer/blocks.py
+++ b/trellis/modules/sparse/transformer/blocks.py
--- a/trellis/modules/transformer/init.py
+++ b/trellis/modules/transformer/init.py
--- a/trellis/modules/transformer/blocks.py
+++ b/trellis/modules/transformer/blocks.py
--- a/trellis/pipelines/init.py
+++ b/trellis/pipelines/init.py
--- a/trellis/pipelines/base.py
+++ b/trellis/pipelines/base.py
--- a/trellis/pipelines/samplers/base.py
+++ b/trellis/pipelines/samplers/base.py
--- a/trellis/pipelines/samplers/classifier_free_guidance_mixin.py
+++ b/trellis/pipelines/samplers/classifier_free_guidance_mixin.py
--- a/trellis/representations/mesh/init.py
+++ b/trellis/representations/mesh/init.py
--- a/trellis/representations/mesh/cube2mesh.py
+++ b/trellis/representations/mesh/cube2mesh.py
--- a/trellis/representations/mesh/flexicubes/DCO.txt
+++ b/trellis/representations/mesh/flexicubes/DCO.txt
--- a/trellis/trainers/init.py
+++ b/trellis/trainers/init.py
--- a/trellis/trainers/base.py
+++ b/trellis/trainers/base.py
--- a/trellis/trainers/basic.py
+++ b/trellis/trainers/basic.py
--- a/trellis/trainers/flow_matching/mixins/classifier_free_guidance.py
+++ b/trellis/trainers/flow_matching/mixins/classifier_free_guidance.py
--- a/trellis/utils/data_utils.py
+++ b/trellis/utils/data_utils.py
--- a/trellis/utils/dist_utils.py
+++ b/trellis/utils/dist_utils.py