问题导入模块:ModuleNotFoundError:没有名为“ torch.utils.checkpoint”的模块

时间:2019-11-23 21:54:03

标签: python import pytorch google-colaboratory fast-ai

我正在关注ULMFiT的example。这是github

我正在Google Colab中运行model_forward_pass.ipynb。我正在运行第一个代码块:

import pandas as pd
import numpy as np
import torch
!pip install fastai==0.7.0
!pip install torch
import fastai
from fastai.imports import *
from fastai.torch_imports import *
from fastai.core import *
from fastai.model import fit
from fastai.dataset import *

import torchtext
from torchtext import vocab, data
from torchtext.datasets import language_modeling

from fastai.rnn_reg import *
from fastai.rnn_train import *
from fastai.nlp import *
from fastai.lm_rnn import *
from fastai.text import *

import dill as pickle
!pip install spacy
!python -m spacy download en
import spacy
import html

import pickle
import collections
#!pip install nbimporter
import nbimporter

我遇到以下错误:

ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-1-bd3be060f151> in <module>()
      6 import fastai
      7 from fastai.imports import *
----> 8 from fastai.torch_imports import *
      9 from fastai.core import *
     10 from fastai.model import fit

3 frames
/usr/local/lib/python3.6/dist-packages/fastai/torch_imports.py in <module>()
      1 import os
----> 2 import torch, torchvision, torchtext
      3 from torch import nn, cuda, backends, FloatTensor, LongTensor, optim
      4 import torch.nn.functional as F
      5 from torch.autograd import Variable

/usr/local/lib/python3.6/dist-packages/torchvision/__init__.py in <module>()
      1 import warnings
      2 
----> 3 from torchvision import models
      4 from torchvision import datasets
      5 from torchvision import ops

/usr/local/lib/python3.6/dist-packages/torchvision/models/__init__.py in <module>()
      4 from .squeezenet import *
      5 from .inception import *
----> 6 from .densenet import *
      7 from .googlenet import *
      8 from .mobilenet import *

/usr/local/lib/python3.6/dist-packages/torchvision/models/densenet.py in <module>()
      3 import torch.nn as nn
      4 import torch.nn.functional as F
----> 5 import torch.utils.checkpoint as cp
      6 from collections import OrderedDict
      7 from .utils import load_state_dict_from_url

ModuleNotFoundError: No module named 'torch.utils.checkpoint'

我找到了类似的问题here,但是我仍然不确定如何解决问题。 我该怎么做才能解决此问题?

0 个答案:

没有答案