Skip to content

Commit

Permalink
ci
Browse files Browse the repository at this point in the history
  • Loading branch information
distributedstatemachine committed Dec 30, 2024
1 parent 85f1322 commit 8f71cc0
Show file tree
Hide file tree
Showing 3 changed files with 108 additions and 33 deletions.
72 changes: 72 additions & 0 deletions docker-compose-test.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
services:
miner1:
build:
context: .
dockerfile: docker/Dockerfile
container_name: templar-miner-M111
volumes:
- ~/.bittensor/wallets:/root/.bittensor/wallets
- ./logs:/app/logs
environment:
NODE_TYPE: miner
WALLET_NAME: Bistro
WALLET_HOTKEY: M111
CUDA_DEVICE: cuda:0
NETWORK: test
DEBUG: 'true'
WANDB_API_KEY: ${WANDB_API_KEY}
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ['0', '1', '2']
capabilities: [gpu]

miner2:
build:
context: .
dockerfile: docker/Dockerfile
container_name: templar-miner-M222
volumes:
- ~/.bittensor/wallets:/root/.bittensor/wallets
- ./logs:/app/logs
environment:
NODE_TYPE: miner
WALLET_NAME: Bistro
WALLET_HOTKEY: M222
CUDA_DEVICE: cuda:1
NETWORK: test
DEBUG: 'true'
WANDB_API_KEY: ${WANDB_API_KEY}
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ['0', '1', '2']
capabilities: [gpu]

validator:
build:
context: .
dockerfile: docker/Dockerfile
container_name: templar-validator-V11
volumes:
- ~/.bittensor/wallets:/root/.bittensor/wallets
- ./logs:/app/logs
environment:
NODE_TYPE: validator
WALLET_NAME: Bistro
WALLET_HOTKEY: V11
CUDA_DEVICE: cuda:2
NETWORK: test
DEBUG: 'true'
WANDB_API_KEY: ${WANDB_API_KEY}
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ['0', '1', '2']
capabilities: [gpu]
3 changes: 0 additions & 3 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,6 @@ RUN apt-get update && apt-get install -y \
# Install uv
RUN pip install uv

# Install Rust and Cargo
RUN curl https://sh.rustup.rs -sSf | sh -s -- -y

ENV PATH="/root/.cargo/bin:${PATH}"

# Copy project files
Expand Down
66 changes: 36 additions & 30 deletions docker/docker-compose-test.yml
Original file line number Diff line number Diff line change
@@ -1,66 +1,72 @@
services:
miner1:
build: .
build:
context: .
dockerfile: Dockerfile
container_name: templar-miner-M111
volumes:
- ~/.bittensor/wallets:/root/.bittensor/wallets
- ./logs:/app/logs
environment:
- NODE_TYPE=miner
- WALLET_NAME=Bistro
- WALLET_HOTKEY=M111
- CUDA_DEVICE=cuda:0
- NETWORK=test
- DEBUG=true
- WANDB_API_KEY=${WANDB_API_KEY}
NODE_TYPE: miner
WALLET_NAME: Bistro
WALLET_HOTKEY: M111
CUDA_DEVICE: cuda:0
NETWORK: test
DEBUG: 'true'
WANDB_API_KEY: ${WANDB_API_KEY}
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: [ '0', '1', '2' ]
capabilities: [ gpu ]
device_ids: ['0', '1', '2']
capabilities: [gpu]

miner2:
build: .
build:
context: .
dockerfile: Dockerfile
container_name: templar-miner-M222
volumes:
- ~/.bittensor/wallets:/root/.bittensor/wallets
- ./logs:/app/logs
environment:
- NODE_TYPE=miner
- WALLET_NAME=Bistro
- WALLET_HOTKEY=M222
- CUDA_DEVICE=cuda:1
- NETWORK=test
- DEBUG=true
- WANDB_API_KEY=${WANDB_API_KEY}
NODE_TYPE: miner
WALLET_NAME: Bistro
WALLET_HOTKEY: M222
CUDA_DEVICE: cuda:1
NETWORK: test
DEBUG: 'true'
WANDB_API_KEY: ${WANDB_API_KEY}
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: [ '0', '1', '2' ]
capabilities: [ gpu ]
device_ids: ['0', '1', '2']
capabilities: [gpu]

validator:
build: .
build:
context: .
dockerfile: Dockerfile
container_name: templar-validator-V11
volumes:
- ~/.bittensor/wallets:/root/.bittensor/wallets
- ./logs:/app/logs
environment:
- NODE_TYPE=validator
- WALLET_NAME=Bistro
- WALLET_HOTKEY=V11
- CUDA_DEVICE=cuda:2
- NETWORK=test
- DEBUG=true
- WANDB_API_KEY=${WANDB_API_KEY}
NODE_TYPE: validator
WALLET_NAME: Bistro
WALLET_HOTKEY: V11
CUDA_DEVICE: cuda:2
NETWORK: test
DEBUG: 'true'
WANDB_API_KEY: ${WANDB_API_KEY}
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: [ '0', '1', '2' ]
capabilities: [ gpu ]
device_ids: ['0', '1', '2']
capabilities: [gpu]

0 comments on commit 8f71cc0

Please sign in to comment.