diff --git a/.gitignore b/.gitignore old mode 100644 new mode 100755 diff --git a/.gitmodules b/.gitmodules old mode 100644 new mode 100755 diff --git a/.idea/.gitignore b/.idea/.gitignore old mode 100644 new mode 100755 diff --git a/0_glb_to_obj.py b/0_glb_to_obj.py old mode 100644 new mode 100755 diff --git a/0_remesh.py b/0_remesh.py old mode 100644 new mode 100755 diff --git a/1_obj_to_step.py b/1_obj_to_step.py old mode 100644 new mode 100755 diff --git a/2_step_to_svg.py b/2_step_to_svg.py old mode 100644 new mode 100755 diff --git a/A6000部署说明 b/A6000部署说明 new file mode 100755 index 0000000..936237e --- /dev/null +++ b/A6000部署说明 @@ -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') +" \ No newline at end of file diff --git a/CODE_OF_CONDUCT.md b/CODE_OF_CONDUCT.md old mode 100644 new mode 100755 diff --git a/DATASET.md b/DATASET.md old mode 100644 new mode 100755 diff --git a/app.py b/app.py old mode 100644 new mode 100755 diff --git a/app_text.py b/app_text.py old mode 100644 new mode 100755 diff --git a/blender_glb_to_obj.py b/blender_glb_to_obj.py old mode 100644 new mode 100755 diff --git a/client.py b/client.py old mode 100644 new mode 100755 diff --git a/dataset_toolkits/build_metadata.py b/dataset_toolkits/build_metadata.py old mode 100644 new mode 100755 diff --git a/dataset_toolkits/calculate_aesthetic_scores.py b/dataset_toolkits/calculate_aesthetic_scores.py old mode 100644 new mode 100755 diff --git a/dataset_toolkits/datasets/3D-FUTURE.py b/dataset_toolkits/datasets/3D-FUTURE.py old mode 100644 new mode 100755 diff --git a/dataset_toolkits/datasets/ABO.py b/dataset_toolkits/datasets/ABO.py old mode 100644 new mode 100755 diff --git a/trellis/datasets/__init__.py b/trellis/datasets/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/datasets/components.py b/trellis/datasets/components.py old mode 100644 new mode 100755 diff --git a/trellis/models/__init__.py b/trellis/models/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/models/structured_latent_vae/__init__.py b/trellis/models/structured_latent_vae/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/models/structured_latent_vae/base.py b/trellis/models/structured_latent_vae/base.py old mode 100644 new mode 100755 diff --git a/trellis/models/structured_latent_vae/decoder_gs.py b/trellis/models/structured_latent_vae/decoder_gs.py old mode 100644 new mode 100755 diff --git a/trellis/models/structured_latent_vae/decoder_mesh.py b/trellis/models/structured_latent_vae/decoder_mesh.py old mode 100644 new mode 100755 diff --git a/trellis/models/structured_latent_vae/decoder_rf.py b/trellis/models/structured_latent_vae/decoder_rf.py old mode 100644 new mode 100755 diff --git a/trellis/modules/sparse/transformer/__init__.py b/trellis/modules/sparse/transformer/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/modules/sparse/transformer/blocks.py b/trellis/modules/sparse/transformer/blocks.py old mode 100644 new mode 100755 diff --git a/trellis/modules/transformer/__init__.py b/trellis/modules/transformer/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/modules/transformer/blocks.py b/trellis/modules/transformer/blocks.py old mode 100644 new mode 100755 diff --git a/trellis/pipelines/__init__.py b/trellis/pipelines/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/pipelines/base.py b/trellis/pipelines/base.py old mode 100644 new mode 100755 diff --git a/trellis/pipelines/samplers/base.py b/trellis/pipelines/samplers/base.py old mode 100644 new mode 100755 diff --git a/trellis/pipelines/samplers/classifier_free_guidance_mixin.py b/trellis/pipelines/samplers/classifier_free_guidance_mixin.py old mode 100644 new mode 100755 diff --git a/trellis/representations/mesh/__init__.py b/trellis/representations/mesh/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/representations/mesh/cube2mesh.py b/trellis/representations/mesh/cube2mesh.py old mode 100644 new mode 100755 diff --git a/trellis/representations/mesh/flexicubes/DCO.txt b/trellis/representations/mesh/flexicubes/DCO.txt old mode 100644 new mode 100755 diff --git a/trellis/trainers/__init__.py b/trellis/trainers/__init__.py old mode 100644 new mode 100755 diff --git a/trellis/trainers/base.py b/trellis/trainers/base.py old mode 100644 new mode 100755 diff --git a/trellis/trainers/basic.py b/trellis/trainers/basic.py old mode 100644 new mode 100755 diff --git a/trellis/trainers/flow_matching/mixins/classifier_free_guidance.py b/trellis/trainers/flow_matching/mixins/classifier_free_guidance.py old mode 100644 new mode 100755 diff --git a/trellis/utils/data_utils.py b/trellis/utils/data_utils.py old mode 100644 new mode 100755 diff --git a/trellis/utils/dist_utils.py b/trellis/utils/dist_utils.py old mode 100644 new mode 100755