“ACESS NVIDIA da Docker Compose” Respostas de código

ACESS NVIDIA da 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 da 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

Respostas semelhantes a “ACESS NVIDIA da Docker Compose”

Perguntas semelhantes a “ACESS NVIDIA da Docker Compose”

Procure respostas de código populares por idioma

Procurar outros idiomas de código