반응형
If you have Docker 19.03 or later, a typical command to launch the container is:
$ docker run --gpus all -it --rm -v local_dir:container_dir nvcr.io/nvidia/tensorflow:xx.xx-tfx-py3
If you have Docker 19.02 or earlier, a typical command to launch the container is:
$ nvidia-docker run -it --rm -v local_dir:container_dir nvcr.io/nvidia/tensorflow:xx.xx-tfx-py3
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/tensorflow
TensorFlow | NVIDIA NGC
TensorFlow is an open source platform for machine learning. It provides comprehensive tools and libraries in a flexible architecture allowing easy deployment across a variety of platforms and devices.
catalog.ngc.nvidia.com
https://docs.nvidia.com/deeplearning/frameworks/support-matrix/index.html
Frameworks Support Matrix :: NVIDIA Deep Learning Frameworks Documentation
docs.nvidia.com
반응형