pytorch报错RuntimeError: DataLoader worker (pid(s) xxx) exited unexpectedly

Iris_wondering

已于 2024-01-15 12:15:43 修改

阅读量1w

点赞数 7

分类专栏： pytorch 文章标签： pytorch 人工智能 python

于 2020-06-25 17:08:40 首次发布

本文链接：https://blog.csdn.net/Shirelle_/article/details/106960018

版权

pytorch 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

报错

pytorch报错RuntimeError: 
DataLoader worker (pid(s) ... exited unexpectedly

解决

因为torch.utils.data.DataLoader中设置了num_works=4，也就是多线程读取。

根据其他方法，我设置num_works=1或者num_works=0都不行。

最后重装了pytorch，就可以了。

pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

原因分析

过程中显示下面内容，所以，可能是因为有些包用pip install的时候没装上吧。

The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    certifi-2020.6.20          |           py36_0         160 KB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    numpy-base-1.18.5          |   py36hde5b4d6_0         5.1 MB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    mkl_random-1.1.1           |   py36h0573a6f_0         382 KB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    mkl_fft-1.1.0              |   py36h23d657b_0         159 KB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    numpy-1.18.5               |   py36ha1c710e_0           4 KB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
    ------------------------------------------------------------
                                           Total:         5.8 MB

The following NEW packages will be INSTALLED:

    blas:           1.0-mkl                               https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    cudatoolkit:    10.1.243-h6bb024c_0                   https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    intel-openmp:   2020.1-217                            https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    libgfortran-ng: 7.3.0-hdf63c60_0                      https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    mkl:            2020.1-217                            https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    mkl-service:    2.3.0-py36he904b0f_0                  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    mkl_fft:        1.1.0-py36h23d657b_0                  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    mkl_random:     1.1.1-py36h0573a6f_0                  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    ninja:          1.9.0-py36hfd86e86_0                  https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    numpy:          1.18.5-py36ha1c710e_0                 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    numpy-base:     1.18.5-py36hde5b4d6_0                 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main    
    pytorch:        1.4.0-py3.6_cuda10.1.243_cudnn7.6.3_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
    six:            1.15.0-py_0                           https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main 
    ```