这篇博客详细记录了PyTorch编程中遇到的五个典型错误：1) 图像加载大小不一致；2) BrokenPipeError在多线程环境中的问题；3) CUDA内存不足的问题；4) 如何实时监控GPU状态；5) Visdom连接失败的解决方法。通过这些错误分析和解决方案，可以帮助读者更好地理解和避免类似问题。
PyTorch代码错误笔记一
文章目录
- PyTorch代码错误笔记一
- 前言
- 1、RuntimeError: stack expects each tensor to be equal size, but got [1, 224, 224] at entry 0 and [3, 224, 224] at entry 32
- 2、BrokenPipeError: [Errno 32] Broken pipe
- 3、RuntimeError: CUDA out of memory. Tried to allocate 392.00 MiB (GPU 0; 4.00 GiB total capacity; 2.18 GiB already allocated; 381.14 MiB free; 2.20 GiB reserved in total by PyTorch)
- 4、训练网络时，怎么实时查看GPU的名称，温度，显存利用情况和GPU利用率，还有正在工作的GPU进程。
- 5、requests.exceptions.ConnectionError: HTTPConnectionPool(host='localhost', port=8097): Max retries exceeded with url: /env/vvv (Caused by NewConnectionError('
`在这里插入代码片`
前言
本文记录着在学习PyTorch过程中遇到的bug及解决方法。随着学习的不断深入，bug记录也会越来越多。
1、RuntimeError: stack expects each tensor to be equal size, but got [1, 224, 224] at entry 0 and [3, 224, 224] at entry 32
错误分析：代码错误代码为：PIL.Image.open(‘path)，用来加载图片，但是数据集中不仅有三通道图片，还有单通道图像。
解决方案：在加载图片的时候，把图片转化为单通道或者三通道，正确代码为：PIL. Image.open(path).convert(‘L’)
2、BrokenPipeError: [Errno 32] Broken pipe
错误分析：num_workers=2时，由于windows系统下的多线程存在bug吧，有时候可以，有时候不可以，用linux
最低0.47元/天 解锁文章
扫一扫
528
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
