随笔分类 - pytorch

python中利用pipreqs查询并安装项目所依赖的所有包

摘要：pipreqs的安装 pip install pipreqs -i http://pypi.douban.com/simple --trusted-host pypi.douban.com 在terminal中，如要查看test文件夹下脚本所依赖包，则输入如下命令： pipreqs ./test - 阅读全文

posted @ 2022-11-23 15:53 poowicat 阅读(193) 评论(0) 推荐(0) 编辑

解决训练yolor时，出现错误：‘Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!’

摘要：问题 ‘Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!’ 解决方法1：x.to(device) 把 device 作为一个可变参数，推荐使用argparse 阅读全文

posted @ 2022-06-13 15:08 poowicat 阅读(4664) 评论(0) 推荐(0) 编辑