- 博客(8)
- 收藏
- 关注
原创 使用python setup.py install后import包出现ModuleNotFoundError
直接使用python setup.py install为本地安装,安装的路径没有添加到环境变量中,需要执行。或者使用python setup.py install时查看安装路径。以安装torch2trt为例,使用源码安装。如果使用torch2trt。解决方案为找到安装包的位置。最后添加到环境变量中。
2023-06-13 16:27:28
1277
1
原创 docker中实现硬件加速
最近想在docker容器里面使用英伟达的GPU实现硬件加速显示ros的图形界面,但是输入如下代码实现的只是软件加速显示图形界面,查了很久资料终于找到一种docker实现硬件加速的办法。我们可以使用osrf/rocker辅助创建一个可以实现硬件加速的容器。如果显示有nvidia说明能够实现硬件加速。
2023-04-06 11:09:54
441
原创 vscode分布式训练debug
使用多个gpu训练神经网络是,如何配置launch文件来debug是必不可少的。平常分布式训练时,我们一般会写一个shell脚本来配置训练参数,通过torch.distributed.launch来启动分布式训练。debug中使用分布式训练需要找到torch分布式包中的launch.py文件,将launch.json文件中的"program"修改为launch.py的路径。其他参数添加到“args”中。
2022-11-19 17:12:58
879
原创 在vscode中调试python代码
vscode是一款非常好用的编辑器,在vscode中调试python代码也非常简单,只需要修改.vscode文件夹下的launch.json就可以了。
2022-11-19 16:34:36
4247
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人