网站建设与制作石家庄,云南网络公司,c 网站开发入门视频教程,网站代码优化方案目录 一、docker 配置 gpu版pyhorch环境1、显卡驱动、cuda版本、pytorch cuda版本三者对应2、拉取镜像 二、部署Anomalib1、下载Anomalib2、创建容器并且运行3、安装Anomalib进入项目路径安装依赖测试#xff1a; 一、docker 配置 gpu版pyhorch环境
1、显卡驱动、cuda版本、p… 目录 一、docker 配置 gpu版pyhorch环境1、显卡驱动、cuda版本、pytorch cuda版本三者对应2、拉取镜像 二、部署Anomalib1、下载Anomalib2、创建容器并且运行3、安装Anomalib进入项目路径安装依赖测试 一、docker 配置 gpu版pyhorch环境
1、显卡驱动、cuda版本、pytorch cuda版本三者对应
自行查看三者对应版本信息
2、拉取镜像 直接在docker.hub中拉取pytorch镜像找到自己所需要的环境 pytorch镜像 点击复制 devel 版链接 docker pull pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel二、部署Anomalib
1、下载Anomalib
Anomalib github 地址
2、创建容器并且运行 我们需要docker容器中运行Anomalib所以需要把主机文件挂载到容器中 主机路径/home/dell/ljn/anomalib
docker run -it --namepytorch_docker \--gpusall \--shm-size64G \ -v /home/dell/ljn/anomalib:/workspace/anomalib \pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel /bin/bash
没有加这一行--shm-size64G在docker中运行pytorch会报错
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm).由于在docker镜像中默认限制了shmshared memory,然而数据处理时pythorch则使用了shm。这就导致了在运行多线程时会将超出限制的DataLoader并直接被kill掉。 --shm-size64G这里设置为64G根据自身电脑性能进行设置容器运行后可以看到挂载的目录
jnljn-Alienware-13:~/ai_project/anomalib$ ./run_docker.sh
root440738bd07a3:/workspace# ls
anomalib
root440738bd07a3:/workspace# 验证
ncvv -V
nvidia-smi3、安装Anomalib
进入项目路径
cd anomalib/anomalib-main安装依赖
pip install update
pip install -e .
pip install wandb测试 执行 python tools/train.py 运行报错ImportError: libGL.so.1: cannot open shared object file: No such file or directory 需要安装opencv apt-get update
apt-get install libopencv-dev再次执行python tools/train.py 报错ImportError: cannot import name KeypointRCNN_ResNet50_FPN_Weights from torchvision.models.detection (/opt/conda/lib/python3.7/site-packages/torchvision/models/detection/__init__.py) torchvision 版本不兼容或缺少相关模块导致的更新版本 pip install --upgrade torchvision再次执行python tools/train.py 运行成功到此Anomalib项目部署完成
