You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
675 B
YAML
24 lines
675 B
YAML
#
|
|
# Copyright (C) 2018 Intel Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
#
|
|
version: "2.3"
|
|
|
|
services:
|
|
cvat:
|
|
build:
|
|
context: .
|
|
args:
|
|
CUDA_SUPPORT: "yes"
|
|
runtime: "nvidia"
|
|
environment:
|
|
NVIDIA_VISIBLE_DEVICES: all
|
|
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
|
# That environment variable is used by the Nvidia Container Runtime.
|
|
# The Nvidia Container Runtime parses this as:
|
|
# :space:: logical OR
|
|
# ,: Logical AND
|
|
# https://gitlab.com/nvidia/container-images/cuda/issues/31#note_149432780
|
|
NVIDIA_REQUIRE_CUDA: "cuda>=10.0 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=410,driver<411"
|