由于服务器架构不同,需要重新构建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')
"