When I want to try to test the code, I encountered the following problem, how should I solve it?
Here's the error.
Traceback (most recent call last):
File "test.py", line 16, in
import lightning.pytorch as pl
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/init.py", line 25, in
from lightning.fabric.fabric import Fabric # noqa: E402
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/init.py", line 29, in
from lightning.fabric.fabric import Fabric # noqa: E402
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/fabric.py", line 32, in
from lightning.fabric.plugins import Precision # avoid circular imports: # isort: split
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/init.py", line 18, in
from lightning.fabric.plugins.precision.amp import MixedPrecision
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/precision/init.py", line 14, in
from lightning.fabric.plugins.precision.amp import MixedPrecision
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/precision/amp.py", line 29, in
class MixedPrecision(Precision):
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/precision/amp.py", line 96, in MixedPrecision
def _autocast_context_manager(self) -> torch.autocast:
AttributeError: module 'torch' has no attribute 'autocast'
When I want to try to test the code, I encountered the following problem, how should I solve it? Here's the error. Traceback (most recent call last): File "test.py", line 16, in
import lightning.pytorch as pl
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/init.py", line 25, in
from lightning.fabric.fabric import Fabric # noqa: E402
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/init.py", line 29, in
from lightning.fabric.fabric import Fabric # noqa: E402
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/fabric.py", line 32, in
from lightning.fabric.plugins import Precision # avoid circular imports: # isort: split
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/init.py", line 18, in
from lightning.fabric.plugins.precision.amp import MixedPrecision
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/precision/init.py", line 14, in
from lightning.fabric.plugins.precision.amp import MixedPrecision
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/precision/amp.py", line 29, in
class MixedPrecision(Precision):
File "/home/isalab102/anaconda3/envs/promptir/lib/python3.8/site-packages/lightning/fabric/plugins/precision/amp.py", line 96, in MixedPrecision
def _autocast_context_manager(self) -> torch.autocast:
AttributeError: module 'torch' has no attribute 'autocast'