pytorch学习

pytorch常用代码

• 在pytorch中使用tensorboard: https://www.pytorchtutorial.com/pytorch-builtin-tensorboard/
• pytorch常用代码段合集： https://zhuanlan.zhihu.com/p/59205847

pytorch踩坑记录

1. ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm)
   • 出现这个错误的情况是，在服务器上的docker中运行训练代码时，batch size设置得过大，shared memory不够（因为docker限制了shm）.解决方法是，将Dataloader的num_workers设置为0.
   • 参考：https://zhuanlan.zhihu.com/p/59271905
2. AttributeError: module ‘scipy.misc’ has no attribute ‘toimage’

   • 解决方法：由于安装的scipy版本是1.4版本，该版本删除了toimage函数，最方便的解决办法降版本
     

     $ pip uninstall scipy
     $ pip install scipy==1.2.0
3. ModuleNotFoundError: No module named ‘tkinter’
   • 在utils.py中添加以下代码：
     import matplotlib
     matplotlib.use(‘agg’)
     import matplotlib.pyplot as plt