TensorFlow release images to your machine: docker pull tensorflow/tensorflow # latest stable release docker pull tensorflow/tensorflow:devel-gpu # nightly dev release w/ GPU support docker pull tensorflow/tensorflow:latest-gpu-jupyter # latest release w/ GPU support and Jupyter Start a TensorFlow Docker container The specified tag release with Jupyter (includes TensorFlow tutorial notebooks) 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
0 kommentar(er)
