想用GPU跑yolo数据集,但是始终使用的是cpu,运行环境调整好了,还是有报错Memory Management and PYTORCH_CUDA_ALLOC_CONF
使用GPU训练模型,遇到此问题一Memory Management and PYTORCH_CUDA_ALLOC_CONF,问题二[Errno 32] Broken pipe 问题三RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu),这里都有体现,可以参考一下
原创
2023-02-24 17:40:50 ·
4322 阅读 ·
1 评论