flashlight-uk / docker-compose-gpu.yml

Create docker-compose-gpu.yml

e433446 verified over 1 year ago

231 Bytes

version: '3'

services:
  flashlight_gpu:
    image: flml/flashlight:cuda-latest
    runtime: nvidia
    tty: true
    stdin_open: true
    container_name: flashlight_gpu
    volumes:
      - ./data/:/data
      - ./models/:/models

	version: '3'

	services:
	flashlight_gpu:
	image: flml/flashlight:cuda-latest
	runtime: nvidia
	tty: true
	stdin_open: true
	container_name: flashlight_gpu
	volumes:
	- ./data/:/data
	- ./models/:/models