- 博客(8)
- 收藏
- 关注
RSS订阅原创 使用python setup.py install后import包出现ModuleNotFoundError
直接使用python setup.py install为本地安装，安装的路径没有添加到环境变量中，需要执行。或者使用python setup.py install时查看安装路径。以安装torch2trt为例，使用源码安装。如果使用torch2trt。解决方案为找到安装包的位置。最后添加到环境变量中。
2023-06-13 16:27:28
1546
1
原创 docker中实现硬件加速
最近想在docker容器里面使用英伟达的GPU实现硬件加速显示ros的图形界面，但是输入如下代码实现的只是软件加速显示图形界面，查了很久资料终于找到一种docker实现硬件加速的办法。我们可以使用osrf/rocker辅助创建一个可以实现硬件加速的容器。如果显示有nvidia说明能够实现硬件加速。
2023-04-06 11:09:54
562
原创 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
1015
原创 在vscode中调试python代码
vscode是一款非常好用的编辑器，在vscode中调试python代码也非常简单，只需要修改.vscode文件夹下的launch.json就可以了。
2022-11-19 16:34:36
4415
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人