Hi, I'm trying to register a flow and getting pytorch error. Any input is greatly appreciated.
@Haytham Abuelfutuh @Ketan (kumare3)@Samhita Alla
(XADM-1068/Code$ pyflyte register --image openaimp/flyte:1
Traceback (most recent call last):
File "/home/xaimpl/.local/bin/pyflyte", line 5, in <module>
from flytekit.clis.sdk_in_container.pyflyte import main
File "/home/xaimpl/.local/lib/python3.10/site-packages/flytekit/__init__.py", line 185, in <module>
from flytekit.extras import pytorch
File "/home/xaimpl/.local/lib/python3.10/site-packages/flytekit/extras/pytorch/__init__.py", line 26, in <module>
from .checkpoint import PyTorchCheckpoint, PyTorchCheckpointTransformer
File "/home/xaimpl/.local/lib/python3.10/site-packages/flytekit/extras/pytorch/checkpoint.py", line 25, in <module>
class PyTorchCheckpoint:
File "/home/xaimpl/.local/lib/python3.10/site-packages/flytekit/extras/pytorch/checkpoint.py", line 30, in PyTorchCheckpoint
module: Optional[torch.nn.Module] = None
AttributeError: module 'torch' has no attribute 'nn'