va1shn9v / PromptIR

PromptIR: Prompting for All-in-One Blind Image Restoration [NeurIPS 2023]
https://arxiv.org/abs/2306.13090
Other
297 stars 22 forks source link

AttributeError: module 'torch' has no attribute 'autocast' #19

Closed ndmm-zwm closed 3 months ago

ndmm-zwm commented 6 months ago

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'

Threegood-student commented 6 months ago

I have encountered the same issue. Have you resolved it?