pytorch
heroacool
疯狂创造价值!研究兴趣:自动标注、SLAM、点云检测跟踪&分割、WES测序和变异分析。有相关需求可以私信!
展开
-
RuntimeError: CUDA out of memory. Tried to allocate 6.38 GiB (GPU 0; 10.76 GiB total capacity; 708.0
用nvidia-smi查看显存看pid 5466是什么线程ps aux | grep 5466原来是chrome,退出chrome即可正常运行pytorch原创 2021-07-07 14:49:42 · 827 阅读 · 0 评论 -
detectron2检测
detectron2训练自有数据做检测原创 2021-06-23 11:44:52 · 241 阅读 · 0 评论 -
从源码编译detectron2
背景:升级了pytorch到1.9,detectron2没有pytorch1.9对应的预编译好的包,所以需要从源码编译。系统环境:ubuntu16.04步骤安装torchconda create -n conda37 python=3.7conda activate conda37pip install torch torchvision torchaudio编译detectron2git clone https://github.com/facebookresearch/de原创 2021-06-17 17:09:11 · 444 阅读 · 0 评论 -
pytorch报错RuntimeError: DataLoader worker (pid 18906) is killed by signal: Segmentation fault.
系统环境:ubuntu16.04cuda11torch包的版本torch 1.8.1torchaudio 0.8.1torchvision 0.9.1运行官方的tv-training-code.py时出现错误RuntimeError: DataLoader worker (pid 18906) is killed by signal: Segmentation fault.找了半天应该是dat原创 2021-06-17 15:44:54 · 2864 阅读 · 3 评论 -
pytorch问题记录
一遍一遍的看,问题数量逐渐收敛。model.roi_heads是什么?原创 2021-04-07 16:29:52 · 168 阅读 · 0 评论