Channels
#flyte-deployment
Title
# flyte-deployment
o
Open AIMP
10/04/2022, 3:32 AMHi, 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's
Samhita Alla
10/04/2022, 5:24 AM
@Open AIMP, have you created a torch.py file?
y
Yee
10/04/2022, 4:31 PM
do you not have pytorch installed?
Copy code
>>> import torch
>>> type(torch.nn)
<class 'module'>
>>> type(torch.nn.Module)
<class 'type'>o
Open AIMP
10/04/2022, 9:38 PM@Samhita Alla 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?
s
Samhita Alla
10/05/2022, 5:39 AM
Is
torch.nno
Open AIMP
10/06/2022, 6:27 AMyes
s
Samhita Alla
10/06/2022, 6:36 AM
It works for me, @Open AIMP. 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.pyIf the error persists, we can connect on a call.
3 Views