search

3DTopia / OpenLRM

An open-source impl. of Large Reconstruction Models
Apache License 2.0
972 stars 56 forks source link

AttributeError: Float8_e5m2 #56

Closed Huangkangxian closed 3 weeks ago

Huangkangxian commented 3 months ago

Traceback (most recent call last): File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/launch.py", line 21, in from openlrm.runners import REGISTRY_RUNNERS File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/runners/init.py", line 20, in from .train import File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/runners/train/init.py", line 16, in from .lrm import LRMTrainer File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/runners/train/lrm.py", line 21, in from torchvision.utils import make_grid File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchvision/init.py", line 10, in from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchvision/_meta_registrations.py", line 4, in import torch._custom_ops File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torch/_custom_ops.py", line 4, in from torch._custom_op.impl import ( File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torch/_custom_op/impl.py", line 9, in from torchgen.model import FunctionSchema, OperatorName, SchemaKind, BaseType, ListType, BaseTy File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/model.py", line 2354, in import torchgen.api.ufunc as ufunc File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/api/ufunc.py", line 11, in import torchgen.api.types as api_types File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/api/types/init.py", line 1, in from .types import File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/api/types/types.py", line 100, in ScalarType.Float8_e5m2: float8_e5m2T, File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/enum.py", line 437, in getattr raise AttributeError(name) from None AttributeError: Float8_e5m2

How should I solve this error?

dongchen-yang commented 3 weeks ago

I think this is caused by a mismatch between the cuda versions for pytorch and xformer. Just make sure they have the same version.