在项目中遇到的一个坑,排查了很久,现在记录下来,该问题是由mkldnn和pytorch版本共同引发,解决方案如下:
首先pytorch版本变更为1.3.1,
然后在代码中添加:
from torch.utils import mkldnn
mkldnn.enabled = False
os.environ['LRU_CACHE_CAPACITY'] = '1'
在项目中遇到的一个坑,排查了很久,现在记录下来,该问题是由mkldnn和pytorch版本共同引发,解决方案如下:
首先pytorch版本变更为1.3.1,
然后在代码中添加:
from torch.utils import mkldnn
mkldnn.enabled = False
os.environ['LRU_CACHE_CAPACITY'] = '1'