ImportError: cannot import name ‘IterableDataset‘ from ‘torch.utils.data.dataset‘

今天在导入torch库的时候出现了下面的错误：

ImportError: cannot import name 'IterableDataset' from 'torch.utils.data.dataset'

我的代码是：

import json
import pandas as pd
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import Dataset, DataLoader
# from torch.cuda.amp import autocast, GradScaler
from tqdm import tqdm
import random
import os
import time
from sklearn.model_selection import *
from transformers import *

解决方法

pip install torch==1.5

输出日志：

Looking in indexes: http://pip.baidu.com/root/baidu/+simple/
Collecting torch==1.5
  Downloading http://pip.baidu.com/root/baidu/packages/packages/59/6f/cdee668c94f5efb3745e9485765fd6b4918a855f7d36c0514ddf38daaddf/torch-1.5.0-cp37-none-macosx_10_9_x86_64.whl (80.5MB)
     |████████████████████████████████| 80.5MB 12.0MB/s
Requirement already satisfied: future in /Users/wugaosheng/anaconda3/lib/python3.7/site-packages (from torch==1.5) (0.18.2)
Requirement already satisfied: numpy in /Users/wugaosheng/anaconda3/lib/python3.7/site-packages (from torch==1.5) (1.19.5)
ERROR: torchnlp 0.0.0.1 has requirement torch==1.0.0, but you'll have torch 1.5.0 which is incompatible.
Installing collected packages: torch
  Found existing installation: torch 1.0.0
    Uninstalling torch-1.0.0:
      Successfully uninstalled torch-1.0.0
Successfully installed torch-1.5.0

参考文献

FARMReader on Conda Environment