xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2025-12-24 12:14:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge f5f8ab2550380c5f223f5a9c34e209aac187c813 into 89536cbd424114c843fdade6f2832e444d0f2b20

Browse Source
This commit is contained in:
AustinMroz 2025-02-05 16:38:15 +00:00 committed by GitHub
commit a15467407b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
hy3dgen/texgen/custom_rasterizer/setup.py
View File

@ -3,9 +3,12 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension
import torch
torch_version = torch.__version__.split('+')[0].replace('.', '')
cuda_version = torch.version.cuda.replace('.', '')
if torch.version.cuda is not None:
cuda_version = "cuda"+torch.version.cuda.replace('.', '')
elif getattr(torch.version, 'hip', None) is not None:
cuda_version = "rocm" + torch.version.hip.replace('.','')
version = f"0.1.0+torch{torch_version}.cuda{cuda_version}"
version = f"0.1.0+torch{torch_version}.{cuda_version}"
# build custom rasterizer
# build with `python setup.py install`
# nvcc is needed