Hi, I'm trying to register a flow and getting pyto...
# flyte-deploymentn
nutritious-flower-32708
10/04/2022, 3:32 AMHi, I'm trying to register a flow and getting pytorch error. Any input is greatly appreciated. @high-park-82026 @freezing-airport-6809@tall-lock-23197
(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'🖖 1
t
tall-lock-23197
10/04/2022, 5:24 AM
@nutritious-flower-32708, have you created a torch.py file?
t
thankful-minister-83577
10/04/2022, 4:31 PM
do you not have pytorch installed?
thankful-minister-83577
10/04/2022, 4:31 PM
Copy code
>>> import torch
>>> type(torch.nn)
<class 'module'>
>>> type(torch.nn.Module)
<class 'type'>n
nutritious-flower-32708
10/04/2022, 9:38 PM@tall-lock-23197 no, I've not created a torch.py file. Can you guide me or share a sample file? Also, where that file needs to go?
nutritious-flower-32708
10/04/2022, 9:41 PM@thankful-minister-83577 I've pytorch installed
t
tall-lock-23197
10/05/2022, 5:39 AM
Is
torch.nnn
nutritious-flower-32708
10/06/2022, 6:27 AMyes
t
tall-lock-23197
10/06/2022, 6:36 AM
It works for me, @nutritious-flower-32708. Also I noticed you’re not giving a package or a module to pyflyte register command. It has to be of this format:
pyflyte register --image <http://ghcr.io/flyteorg/flytecookbook:core-latest|ghcr.io/flyteorg/flytecookbook:core-latest> pytorch_types.pytall-lock-23197
10/06/2022, 6:37 AM
If the error persists, we can connect on a call.
168 Views