I tried importing functorch and torch in python 3.9.7
import torch
from functorch import vmap, grad
and ran into a circular import issue.
---------------------------------------------------------------------------
ImportError Traceback (most recent call last)
/var/folders/4p/5v2n8wpn58v6xw674tjlss680000gn/T/ipykernel_64234/1530828949.py in <module>
1 import torch
----> 2 from functorch import vmap, grad
/usr/local/lib/python3.11/site-packages/functorch/__init__.py in <module>
5 # LICENSE file in the root directory of this source tree.
6 import torch
----> 7 from . import _C
8
9 # Top-level APIs. Please think carefully before adding something to the
ImportError: cannot import name '_C' from partially initialized module 'functorch' (most likely due to a circular import) (/usr/local/lib/python3.11/site-packages/functorch/__init__.py)
Any way around it? Is this a bug in the packages?
Submitting this here since I could not find an easy fix