- 博客(5)
- 收藏
- 关注
原创 关于docker使用GPU的一些官方操作 nvidia-container-runtime
3.操作完yum或者apt-get install nvidia-container-runtime 如果是windows按照windows方式下载。2.根据系统进行复制粘贴操作即可。4.重启docker(非常必要)
2023-10-27 17:19:52 294 1
原创 tensorrt和onnxruntime-gpu同时调用gpu时tensorrt推理出现错乱解决方式
tensorrt和onnxruntime-gpu同进程冲突
2023-03-02 17:54:26 984
原创 pytorch半精度浮点训练踩坑
1.loss,最开始因为是u2netp跑的分割,采用的bceloss,然后会cuda核函数运算错误,暂时没有尝试用pycuda去处理,使用mseloss就能完美运行了,大家也可以自己写loss又或者换其他loss试试。2.优化器,最开始用的adam优化器,但是万能的adam出毛病了,梯度爆炸了,采用默认参数依旧梯度爆炸,最开始还以为没有加激活函数,也查了bn,但是没什么问题,换成sgd+余弦褪火就不会产生梯度爆炸的问题了。
2022-11-07 19:15:52 579
原创 numpy 更新版本后 more than 1 DLL from .libs警告
numpy more than 1 DLL from .libs警告
2022-06-10 13:26:56 2959 4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人