“ACESS NVIDIA de Docker Compose” Código de respuesta

ACESS NVIDIA de Docker Compose

# Add the following line to docker-compose

services:
  test:
    image: tensorflow/tensorflow:latest-gpu
    command: python -c "import tensorflow as tf;tf.test.gpu_device_name()"
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
Merwanski

ACESS NVIDIA de Docker Compose


test_1  | 2021-01-23 11:02:46.500189: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1402] Created TensorFlow device (/device:GPU:0 with 1624 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1050, pci bus id: 0000:01:00.0, compute capability: 6.1)

Sparkling Skimmer

Respuestas similares a “ACESS NVIDIA de Docker Compose”

Preguntas similares a “ACESS NVIDIA de Docker Compose”

Explore las respuestas de código populares por idioma

Explorar otros lenguajes de código