要獲取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運行時。請根據您的需求選擇合適的版本。