Pytorch 0.4.0 基于Windows 10 问题解决记录
1.运行Pytorch tutorial代码报错:BrokenPipeError: [Errno 32] Broken pipe
该问题的产生是由于windows下多线程的问题,和DataLoader类有关,具体细节点这里Fix memory leak when using multiple workers on Windows。
解决方案:
修改调用torch.utils.data.DataLoader()函数时的 num_workers 参数。该参数官方API解释如下:
|
该参数是指在进行数据集加载时,启用的线程数目。截止当前2018年5月9日11:15:52,如官方未解决该BUG,则可以通过修改num_works参数为 0 ,只启用一个主进程加载数据集,避免在windows使用多线程即可。
2.开发IDE-Pycharm的一个小问题
描述:不能在当前编辑的py文件中导入同级目录下另外一个py文件的函数、类等(主要可以运行但是不提示!)
解决方法:
(1)勾选下面的选项:
(2)被导入的py文件所在的文件夹上右键如下图设置:
-- updating --