PyTorch库使用常见的问题集锦(不断完善中)

问题1：AttributeError: ‘_MultiProcessingDataLoaderIter’ object has no attribute ‘_send_idx’

• 问题原因：data.DataLoader没有属性’_send_idx’
  出现这个问题，最大的可能性是因为，torch的版本太低，建议升级Pytorch的版本，本人的版本是torch=1.3.0 torchvision=0.4.1
• 解决策略：
  • 方法１：重现安装pytorchpip install torch==1.3.0 torchvision==0.4.1 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
  • 方法２：在https://download.pytorch.org/whl/torch_stable.html网页下载需要的PyTorch 和　torchvision 版本（注意：cuda 的版本和 Python 的版本相对应）
    安装命令：pip install home/jankin/torch-1.3.0-cp35-cp35m-linux_x86_64.whl

相似问题1.1：AssertionError:The NVIDIA driver on your system is too old (found version 8000).

version from the URL: Download Drivers
Alternatively, go to: PyTorch to install
a PyTorch version that has been compiled with your version
of the CUDA driver.

• 问题原因：　PyTorch 与 CUDA 的版本不匹配
• 解决策略：同上述方法。

问题2：RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50

• 问题原因：Pytorch的版本不太合适
• 解决策略：重新安装pytorch>=1.1 <1.3的版本

问题3：RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50

• 问题原因：无法调用本机的cuda。
• 解决策略：（1）查看是否是本机的显卡驱动没有安装好，检测命令nvidia-smi，若是本机的驱动没有安装好，请重新安装驱动。

问题４：AttributeError: ‘DataParallel’ object has no attribute ‘vocab’

• 问题原因：使用多卡训练，无法调用部分模块
• 解决策略：改为 model.module