ParsAi_lora / docker-compose.yml
caarleexx's picture
Create docker-compose.yml
9cf030f verified
raw
history blame contribute delete
533 Bytes
version: '3.8'
services:
jurifob-training:
build: .
container_name: jurifob-phd-training
environment:
- WANDB_API_KEY=${WANDB_API_KEY}
- HF_TOKEN=${HF_TOKEN}
- CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
ports:
- "7860:7860"
volumes:
- ./trabalho:/workspace/trabalho
- ./checkpoints:/workspace/checkpoints
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 8
capabilities: [gpu]
shm_size: 64gb