zoukankan      html  css  js  c++  java
  • fastai v2 windows运行错误解决:RuntimeError: cuda runtime error (801) : operation not supported at .. orch/csrc/generic/StorageSharing.cpp:247

    fastai v2对比v1有极大的改动,发布的版本主要在linux下运行,并经测试。

    windows在运行learn.fine_tune(1)时出错:

    d:programdataanaconda3envsfastailibsite-packages	orchmultiprocessing
    eductions.py in reduce_tensor(tensor)
        238          ref_counter_offset,
        239          event_handle,
    --> 240          event_sync_required) = storage._share_cuda_()
        241         tensor_offset = tensor.storage_offset()
        242         shared_cache[handle] = StorageWeakRef(storage)
    
    RuntimeError: cuda runtime error (801) : operation not supported at ..	orch/csrc/generic/StorageSharing.cpp:247

    解决办法:在ImageDataLoaders中添加:num_workers=0

    dls = ImageDataLoaders.from_folder(path, train='train', valid='test', item_tfms=Resize(224), num_workers=0)

    此外在learn.predict()、interp = Interpretation.from_learner(learn)中也会出现同样的错误,需修改fastailearner.py,添加n_workers = 0:

        def get_preds(self, ds_idx=1, dl=None, with_input=False, with_decoded=False, with_loss=False, act=None,
                      inner=False, reorder=True, cbs=None, n_workers=defaults.cpus, **kwargs):
            n_workers = 0
            if dl is None: dl = self.dls[ds_idx].new(shuffled=False, drop_last=False, num_workers=n_workers)
  • 相关阅读:
    NET导入Excel带进度条。
    直接拿来用,最火的.NET开源项目(beta)
    Darren Ji
    接口和抽象类有什么区别
    wpf博客
    jQuery动态改变图片显示大小(修改版)
    S32K的make过程
    TortoiseGit安装及使用
    Python:tkinter
    GCC学习笔记(二):编译选项
  • 原文地址:https://www.cnblogs.com/zhengbiqing/p/13654973.html
Copyright © 2011-2022 走看看