要获取PyTorch的Docker镜像,请按照以下步骤操作:
首先,确保您已经安装了Docker。如果尚未安装,请访问Docker官方网站(https://www.docker.com/)以获取适用于您的操作系统的安装指南。
打开终端(在Windows上为命令提示符或PowerShell,在macOS和Linux上为终端)。
拉取PyTorch官方Docker镜像。在终端中输入以下命令,然后按Enter键:
docker pull pytorch/pytorch
这将自动从Docker Hub下载PyTorch的最新版本。下载完成后,您可以在本地运行PyTorch容器。
请注意,如果您希望在Docker容器中使用特定的CUDA版本,可以在拉取镜像时指定CUDA版本。例如,要使用CUDA 11.1,请运行以下命令:
docker pull pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
这将下载PyTorch 1.9.0版本,并为其安装CUDA 11.1和cuDNN 8运行时。请根据您的需求选择合适的版本。