首页 > 其他分享 >pytorch使用docker部署后卡死现象

pytorch使用docker部署后卡死现象

时间:2022-11-18 16:23:34浏览次数:51  
标签:服务 CPU pytorch 现象 docker 卡死

现象

基于pytorch的模型服务,本地裸跑代码都是正常的,一旦上docker服务部署后,程序会出现卡死现象

解决

原因是,默认情况下,pytorch会启动宿主机当前的CPU核数作为线程数去运行,一旦该docker服务是和其他服务共享机器,就容易造成CPU占用过高,出现卡死。指定pytorch使用单线程运行可解决。有如下两种方式可解决

  1. export OMP_NUM_THREADS=1
  2. torch.set_num_threads(1)

标签:服务,CPU,pytorch,现象,docker,卡死
From: https://www.cnblogs.com/baiyiqingxiang/p/16903626.html

相关文章

  • 神经网络中的权重初始化方式和pytorch应用
    目录计算增益常数初始化均匀分布初始化正态分布初始化Xavier初始化均匀分布(glorot初始化)正态分布Kaiming初始化均匀分布正态分布具体应用一些问答或tips深度学习模型中的......
  • Docker离线部署Nginx
    总体思路:在有网络的环境上制作Nginx的镜像包,导出并上传至无网络的环境上,启动Nginx即可。  在上一篇《无网环境DockerRpm离线安装》里面,已经在联网的机器上安装好了......
  • Docker安装nginx
    1、拉取nginx镜像并查看#docker搜索nignxdockersearchnginx#拉取nginx镜像--会从docker官方镜像中拉取dockerpullnginx#查看docker镜像dockerps-a2、运......
  • iOS 升级到XCode13以后运行模拟器经常导致MacOS系统卡死
    升级到XCode13后运行模拟器会导致MacOS系统卡死,升级到XCode14后,该问题仍然存在,当然也有可能是公司办公电脑配置太低导致的[dog],卡死频率很高,会导致整个屏幕无法操作,无法看......
  • win10安装cuda、cuDNN和pytorch笔记
    特别注意:由于自己安装时没有做记录,所以下面大部分安装步骤图片都是参考的网络图,但不影响阅读,每一步都讲得很详细1.安装CUDA1.1查看自己显卡最高支持的CUDA版本在桌面......
  • docker-compose文件详解
    Compose和Docker兼容性:Compose文件格式有3个版本,分别为1,2.x和3.x目前主流的为3.x其支持docker1.13.0及其以上的版本常用参数:version#指定comp......
  • 无网环境Docker Rpm离线安装
    总体思路:找一台可以联网的linux,下载docker的RPM依赖包而不进行安装(yumlocalinstall),将所有依赖的rpm环境打包好,再在无网环境中解压逐一安装(rpm:--force--nodeps)。提前......
  • Docker离线安装使用
    离线安装dockerhttps://www.cnblogs.com/yaoyin/p/16532355.htmldocker离线安装并导入镜像https://blog.csdn.net/m0_67266585/article/details/124174122......
  • docker安装mysql8.0
    目的:记录安装过程的命令和遇到的坑!1.拉取mysql8.0官方镜像dockerpullmysql:8.02.启动命令dockerrun--namemysql--restart=always-p3306:3306--privileged=......
  • docker 常用命令
    转自:https://www.runoob.com/docker/docker-command-manual.html1.创建一个新的容器#使用docker镜像nginx:latest以后台模式启动一个容器,并将容器命名为mynginx。doc......