mirror of
https://github.com/Toumash/mlflow-docker
synced 2025-11-04 23:29:19 +01:00
Compare commits
12 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
01e8abe89a | ||
|
|
d0a5dfbde0 | ||
|
|
f92d4ec230 | ||
|
|
2ad2c983db | ||
|
|
b6ecfe7d0c | ||
|
|
fcd3393fa5 | ||
|
|
14df7c707e | ||
|
|
7506d3e43d | ||
|
|
8f3d6ba7e2 | ||
|
|
81e373d6fb | ||
|
|
eebc9d0c46 | ||
|
|
bddbec77f1 |
11
README.md
11
README.md
@@ -4,6 +4,8 @@ If you want to boot up mlflow project with one-liner - this repo is for you.
|
||||
|
||||
The only requirement is docker installed on your system and we are going to use Bash on linux/windows.
|
||||
|
||||
[](https://www.youtube.com/watch?v=ma5lA19IJRA)
|
||||
|
||||
# Features
|
||||
- Setup by one file (.env)
|
||||
- Production-ready docker volumes
|
||||
@@ -24,7 +26,14 @@ Creating tracker_mlflow ... done
|
||||
Creating aws-s3 ... done
|
||||
```
|
||||
|
||||
3. Create mlflow bucket. You can do it **either using AWS CLI or Python Api**. **You dont need an AWS subscription**
|
||||
3. Create mlflow bucket. You can use my bundled script.
|
||||
|
||||
Just run
|
||||
```shell
|
||||
bash ./run_create_bucket.sh
|
||||
```
|
||||
|
||||
You can also do it **either using AWS CLI or Python Api**.
|
||||
<details><summary>AWS CLI</summary>
|
||||
|
||||
1. [Install AWS cli](https://aws.amazon.com/cli/) **Yes, i know that you dont have an Amazon Web Services Subscription - dont worry! It wont be needed!**
|
||||
|
||||
38
create_bucket.py
Normal file
38
create_bucket.py
Normal file
@@ -0,0 +1,38 @@
|
||||
import os
|
||||
|
||||
from minio import Minio
|
||||
from minio.error import InvalidResponseError
|
||||
|
||||
accessID = os.environ.get('AWS_ACCESS_KEY_ID')
|
||||
accessSecret = os.environ.get('AWS_SECRET_ACCESS_KEY')
|
||||
minioUrl = os.environ.get('MLFLOW_S3_ENDPOINT_URL')
|
||||
bucketName = os.environ.get('AWS_BUCKET_NAME')
|
||||
|
||||
if accessID == None:
|
||||
print('[!] AWS_ACCESS_KEY_ID environemnt variable is empty! run \'source .env\' to load it from the .env file')
|
||||
exit(1)
|
||||
|
||||
if accessSecret == None:
|
||||
print('[!] AWS_SECRET_ACCESS_KEY environemnt variable is empty! run \'source .env\' to load it from the .env file')
|
||||
exit(1)
|
||||
|
||||
if minioUrl == None:
|
||||
print('[!] MLFLOW_S3_ENDPOINT_URL environemnt variable is empty! run \'source .env\' to load it from the .env file')
|
||||
exit(1)
|
||||
|
||||
|
||||
if bucketName == None:
|
||||
print('[!] AWS_BUCKET_NAME environemnt variable is empty! run \'source .env\' to load it from the .env file')
|
||||
exit(1)
|
||||
|
||||
minioUrlHostWithPort = minioUrl.split('//')[1]
|
||||
print('[*] minio url: ',minioUrlHostWithPort)
|
||||
|
||||
s3Client = Minio(
|
||||
minioUrlHostWithPort,
|
||||
access_key=accessID,
|
||||
secret_key=accessSecret,
|
||||
secure=False
|
||||
)
|
||||
|
||||
s3Client.make_bucket(bucketName)
|
||||
@@ -1,50 +1,58 @@
|
||||
version: '3.2'
|
||||
version: "3.9"
|
||||
services:
|
||||
s3:
|
||||
image: minio/minio:latest
|
||||
image: minio/minio:RELEASE.2021-08-25T00-41-18Z.hotfix.55efceab5
|
||||
restart: unless-stopped
|
||||
container_name: aws-s3
|
||||
ports:
|
||||
- 9000:9000
|
||||
- "9000:9000"
|
||||
- "9001:9001"
|
||||
environment:
|
||||
- MINIO_ACCESS_KEY=${AWS_ACCESS_KEY_ID}
|
||||
- MINIO_SECRET_KEY=${AWS_SECRET_ACCESS_KEY}
|
||||
command:
|
||||
server /date
|
||||
- MINIO_ROOT_USER=${AWS_ACCESS_KEY_ID}
|
||||
- MINIO_ROOT_PASSWORD=${AWS_SECRET_ACCESS_KEY}
|
||||
command: server /data --console-address ":9001"
|
||||
networks:
|
||||
- A
|
||||
- internal
|
||||
- public
|
||||
volumes:
|
||||
- ./s3:/date
|
||||
- minio_volume:/data
|
||||
db:
|
||||
restart: always
|
||||
image: mysql/mysql-server:5.7.28
|
||||
container_name: mlflow_db
|
||||
expose:
|
||||
- "3306"
|
||||
environment:
|
||||
- MYSQL_DATABASE=${MYSQL_DATABASE}
|
||||
- MYSQL_USER=${MYSQL_USER}
|
||||
- MYSQL_PASSWORD=${MYSQL_PASSWORD}
|
||||
- MYSQL_ROOT_PASSWORD=${MYSQL_ROOT_PASSWORD}
|
||||
volumes:
|
||||
- ./dbdata:/var/lib/mysql
|
||||
networks:
|
||||
- A
|
||||
image: mysql/mysql-server:5.7.28
|
||||
restart: unless-stopped
|
||||
container_name: mlflow_db
|
||||
expose:
|
||||
- "3306"
|
||||
environment:
|
||||
- MYSQL_DATABASE=${MYSQL_DATABASE}
|
||||
- MYSQL_USER=${MYSQL_USER}
|
||||
- MYSQL_PASSWORD=${MYSQL_PASSWORD}
|
||||
- MYSQL_ROOT_PASSWORD=${MYSQL_ROOT_PASSWORD}
|
||||
volumes:
|
||||
- db_volume:/var/lib/mysql
|
||||
networks:
|
||||
- internal
|
||||
mlflow:
|
||||
container_name: tracker_mlflow
|
||||
image: tracker_ml
|
||||
build:
|
||||
context: ./mlflow
|
||||
dockerfile: Dockerfile
|
||||
ports:
|
||||
- "5000:5000"
|
||||
environment:
|
||||
- AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}
|
||||
- AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY}
|
||||
- AWS_DEFAULT_REGION=${AWS_REGION}
|
||||
- MLFLOW_S3_ENDPOINT_URL=http://s3:9000
|
||||
networks:
|
||||
- A
|
||||
entrypoint: ./wait-for-it.sh db:3306 -t 90 -- mlflow server --backend-store-uri mysql+pymysql://${MYSQL_USER}:${MYSQL_PASSWORD}@db:3306/${MYSQL_DATABASE} --default-artifact-root s3://${AWS_BUCKET_NAME}/ -h 0.0.0.0
|
||||
container_name: tracker_mlflow
|
||||
image: tracker_ml
|
||||
restart: unless-stopped
|
||||
build:
|
||||
context: ./mlflow
|
||||
dockerfile: Dockerfile
|
||||
ports:
|
||||
- "5000:5000"
|
||||
environment:
|
||||
- AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}
|
||||
- AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY}
|
||||
- AWS_DEFAULT_REGION=${AWS_REGION}
|
||||
- MLFLOW_S3_ENDPOINT_URL=http://s3:9000
|
||||
networks:
|
||||
- public
|
||||
- internal
|
||||
entrypoint: ./wait-for-it.sh db:3306 -t 90 -- mlflow server --backend-store-uri mysql+pymysql://${MYSQL_USER}:${MYSQL_PASSWORD}@db:3306/${MYSQL_DATABASE} --default-artifact-root s3://${AWS_BUCKET_NAME}/ -h 0.0.0.0
|
||||
networks:
|
||||
A:
|
||||
driver: bridge
|
||||
internal:
|
||||
public:
|
||||
driver: bridge
|
||||
volumes:
|
||||
db_volume:
|
||||
minio_volume:
|
||||
6
run_create_bucket.sh
Normal file
6
run_create_bucket.sh
Normal file
@@ -0,0 +1,6 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -o allexport; source .env; set +o allexport
|
||||
|
||||
pip3 install Minio
|
||||
python3 ./create_bucket.py
|
||||
Reference in New Issue
Block a user