使用docker搭建faiss向量数据库

码道功成

已于 2024-07-15 15:10:44 修改

阅读量2k

点赞数 8

分类专栏：深度学习文章标签： faiss docker

于 2024-03-18 16:21:50 首次发布

本文链接：https://blog.youkuaiyun.com/NDF923/article/details/136811978

版权

本文介绍了如何在CentOS7的Docker镜像基础上，安装Miniconda，创建conda环境并尝试安装faiss-gpu和pandas。遇到问题后，作者通过编写Dockerfile构建镜像，并使用volume映射将数据文件夹挂载到容器中，以便在容器内运行程序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

为了不污染服务器环境，保证程序运行时有更好的隔离性，领导要求基于容器运行程序。

一、准备工作

1、创建文件夹faiss

该文件夹有用于存放faiss相关的文件及脚本

mkdir ~/faiss

2、创建data文件夹

cd ~/faiss

mkdir data

这个文件夹用于volume，映射容器内部文件夹

二、编写Dockerfile

# 使用 centos:7.9 作为基础镜像
FROM centos:7

# 安装必要的系统软件包
RUN yum install -y wget bzip2 vim

# 下载并安装 Miniconda
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh
RUN bash miniconda.sh -b -p /opt/conda
RUN rm miniconda.sh
ENV PATH="/opt/conda/bin:${PATH}"

# 创建 conda 环境并安装 faiss-gpu 和 pandas
RUN conda create -y -n faiss_env python=3.10 && echo "conda activate faiss_env" >> ~/.bashrc 

#RUN conda create -n faiss_env python=3.10
#RUN conda init bash
#RUN conda activate faiss_env && \
#    pip3 install faiss-gpu -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
#RUN pip3 install pandas -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

# 设置工作目录
WORKDIR /app

主要是创建了一个基于centos7的镜像，然后按照miniconda、faiss-gpu 和 pandas，但是奇怪的是在这里不管是执行pip install 还是conda install，都执行不成功，但是把这个放到容器中执行就可以，咱也不知道为啥。

三、构建镜像

1）创建一个名为faiss-gpu