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算法代碼就行了

總結

以上是生活随笔為你收集整理的Docker部署AI算法教程的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。