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.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |