栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Docker部署AI算法教程

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Docker部署AI算法教程

docker上部署算法除了一些推理框架外,有时候会自己用torch推理加上一些web应用,下面写下自己用的一套方法。

Docker+cuda10.1+miniconda3+torch1.7.1

docker要求19.03以上才支持cuda

1.安装docker,我这里使用的centos7

$ sudo yum install -y yum-utils device-mapper-persistent-data lvm2
$ sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
$ sudo yum install docker-ce


启动并加入开机启动
$ sudo systemctl start docker
$ sudo systemctl enable docker

2.安装nvidia-container-runtime

sudo yum install nvidia-container-runtime

3.下载cuda docker镜像

docker pull nvidia/cuda:10.1-base-ubuntu18.04

4.启动一个容器(顺便把/hone目录挂载到容器中的/host 方便访问文件)

sudo docker run -it --gpus all --name train_server_1 -v /home:/host nvidia/cuda

5.在容器中安装miniconda3

下载地址:

https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

sh Miniconda3-latest-Linux-x86_64.sh(一路yes 回车就可以了)

再使其python环境生效::

source ~/.bashrc

5.安装torch

pip install torch==1.7.1

6.运行自己的AI算法代码就行了

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/423159.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号