为啥在jupyter notebook 里运行没问题的代码,放在pycharm里会报错呢?

在notebook上正常,在pycharm报错
import os
os.environ["KMP_DUPLICATE_LIB_OK"]  =  "TRUE"

from torchvision import transforms,utils
from torchvision import datasets
import torch
from matplotlib import pyplot as plt



my_trans = transforms.Compose([
    transforms.RandomResizedCrop(224),
    transforms.RandomHorizontalFlip(),
    transforms.ToTensor()
])

train_data = datasets.ImageFolder('E:/datasets/hymenoptera_data',transform=my_trans)
train_loader =torch.utils.data.DataLoader(train_data,batch_size=8,shuffle=True)


for i_batch,img in enumerate(train_loader):
    if i_batch ==0:
        print(img[1])
        fig = plt.figure()
        grid = utils.make_grid(img[0])
        plt.imshow(grid.numpy().transpose((1,2,0)))
        plt.show()
        utils.save_image(grid,'test01.png')
    break

 

报什么错误啊!如果是模块的原因,我想应该是pycharm没有配置python运行环境的原因吧!