- 博客(2)
- 收藏
- 关注
RSS订阅原创 Libtorch在Win10下的另一种部署方法（不使用CMake）
Libtorch在Win10下的另一种部署方法（不使用CMake）目前网上流传的方法大多都是利用了cmake来部署，例如：Windows使用C++和CMake调用Pytorch1.0模型这种方法可以完成，但是libtorch包和OpenCV类似，都是包含了lib等文件，所以也可以像配置第三方库一样配置：我使用的是VS2019，在官网安装最新的Debug版本的libtorch1.8.0之后新建一个项目，在属性管理器中，配置项目属性包含目录写入libtorch中的include库目录写入li
2021-03-09 22:56:16
834
原创 TensorFlow函数使用完成释放显存问题
TensorFlow函数使用完成释放显存问题使用Tensorflow跑程序，3G显存总是会占满，之后使用Pytorch就会显存不足。Tensorflow部分使用完成，显存不会自动释放，直到程序结束才会释放。我们可以借助多进程来释放内存。下面是我尝试各种办法后，得到的目前最好方法。其余办法不是太慢，就是无法输入多个参数，或者得不到返回值import multiprocessingdef extract_regional_features(rgb_list, model_path): reg
2020-11-29 20:59:13
3260
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人