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