반응형
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/pytorch:xx.xx-py3If 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/pytorch:xx.xx-py3
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch
PyTorch | NVIDIA NGC
PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.
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
반응형