Files
mutation/docker/docker-compose.yml
lingyu zeng d1f25e7110 update
2024-09-17 13:20:08 +08:00

47 lines
1.0 KiB
YAML

version: '3.8'
services:
ubuntumutation:
build:
context: ..
dockerfile: Dockerfile
args:
CACHEBUST: 1
CONDA_ENV_NAME: "mutation"
PYTHON_VERSION: "3.11"
TAG_VERSION: "22.04"
HTTP_PROXY: "http://192.168.1.103:7897"
HTTPS_PROXY: "http://192.168.1.103:7897"
volumes:
- ../:/work
container_name: ubuntumutation
pull_policy: if_not_present
ulimits:
memlock:
soft: -1
hard: -1
restart: unless-stopped
image: hotwa/mutation:testbuild
privileged: true
cap_add:
- ALL
- CAP_SYS_PTRACE
shm_size: '64gb'
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
command: ["/usr/sbin/sshd", "-D"]
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
networks:
- network_ollama
networks:
network_ollama:
name: network_ollama