zagre.blogg.se - Docker run debian

The specified tag release with GPU support. Specify the version of the TensorFlow binary image, for example: 2.8.3Įach base tag has variants that add or change functionality: Tag Variants The latest release of TensorFlow CPU binary image. The official TensorFlow Docker images are located in theĭocker Hub repository. Note: To run the docker command without sudo, create the docker group and

Both options are documented on the page linked above. On versions including and after 19.03, you will use the nvidia-container-toolkit package and the -gpus all flag. Versions earlier than 19.03 require nvidia-docker2 and the -runtime=nvidia flag.

Take note of your Docker version with docker -v.

For GPU support on Linux, install NVIDIA Docker support.

Is required on the host machine (the NVIDIA® CUDA® Toolkit does not need toīe installed). TensorFlow programs are run within this virtual environment thatĬan share resources with its host machine (access directories, use the GPU,ĭocker is the easiest way to enable TensorFlow GPU support on Linux since only the

Create virtual environments that isolate a TensorFlow installation from the rest