move to bgpt dir

This commit is contained in:
2024-06-20 16:20:55 +08:00
parent 06bd0b7f14
commit 3522da961b
2 changed files with 113 additions and 0 deletions

View File

@@ -0,0 +1,32 @@
version: '3.8'
services:
ubuntu-ssh:
build:
context: .
dockerfile: Dockerfile.bgpt
volumes:
- /data:/data
container_name: ubuntu-ssh
pull_policy: if_not_present
tty: true
restart: unless-stopped
image: zly/cuda-bgpt:latest
ports:
- 3222:22
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
networks:
- network_bgpt
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
networks:
network_bgpt:
name: network_bgpt