pip安装PyTorch =1.2、torchVision = 0.3时报错ImportError: DLL load failed: 原因是版本没对！

最新推荐文章于 2025-02-19 22:43:04 发布

快乐成长吧

最新推荐文章于 2025-02-19 22:43:04 发布

阅读量1.2k

点赞数

分类专栏：街景字符编码识别

本文链接：https://blog.youkuaiyun.com/qq_37791134/article/details/106246640

版权

博客内容讲述了在使用pip安装PyTorch 1.2和torchVision 0.3时遇到ImportError: DLL load failed的问题。作者通过排查发现可能是因为版本不匹配导致，然后卸载torchVision 0.3，重新安装了torchVision 0.4.0，成功解决了问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- coding: utf-8 -*-

'''
@Time    : 2020/5/20 22:28
@Author  : HHNa
@FileName: dataset.py
@Software: PyCharm
 
'''

import os, sys, glob, shutil, json
import cv2

from PIL import Image
import numpy as np

import torch
from torch.utils.data.dataset import Dataset
import torchvision.transforms as transforms


class SVHNDataset(Dataset):
    def __init__(self, img_path, img_label, transform=None):
        self.img_path = img_path
        self.img_label = img_label
        if transform is not None:
            self.transform = transform
        else:
            self.transform = None

    def __getitem__(self, index):
        img = Image.open(self.img_path[index]).convert('RGB')

        if self.transform is not None:
            img = self.transform(img)

        # 原始SVHN中类别10为数字0
        lbl = np.array(self.img_label[index], dtype=np.int)
        lbl = list(lbl) + (5 - len(lbl)) * [