1

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')
+"