21 lines
805 B
Plaintext
Executable File
21 lines
805 B
Plaintext
Executable File
由于服务器架构不同,需要重新构建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')
|
|
" |