| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:b0bf3f64519bf10a51e00d4f8ab9c8693620659a0c4a4a5090a9765e754e1d1c in / |
| CMD ["/bin/bash"] |
| ARG VERSION=10.1.0 |
| ARG TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ |
| ARG TORCH_SPECS=torch[device-all]==2.14.0+rocm10.1.0 |
| ARG TORCHVISION_SPECS=torchvision[device-all]==0.29.0a0+rocm10.1.0 |
| ARG TORCHAUDIO_SPECS=torchaudio==2.11.0.3+rocm10.1.0 |
| ARG APEX_SPECS=apex==1.14.0+rocm10.1.0 |
| LABEL org.opencontainers.image.title=ROCm PyTorch (rocm-ai-docker-stack) org.opencontainers.image.description=Multi-arch PyTorch on ROCm; prebuilt wheels from the multi-arch wheel index. org.opencontainers.image.version=10.1.0 com.amd.framework=pytorch com.amd.rocm.gpu-coverage=multi-arch |
| COPY staging/install_rocm_deps.sh /tmp/install_rocm_deps.sh # buildkit |
| RUN |6 VERSION=10.1.0 TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ TORCH_SPECS=torch[device-all]==2.14.0+rocm10.1.0 TORCHVISION_SPECS=torchvision[device-all]==0.29.0a0+rocm10.1.0 TORCHAUDIO_SPECS=torchaudio==2.11.0.3+rocm10.1.0 APEX_SPECS=apex==1.14.0+rocm10.1.0 /bin/sh -c chmod +x /tmp/install_rocm_deps.sh && /tmp/install_rocm_deps.sh && rm -f /tmp/install_rocm_deps.sh # buildkit |
| RUN |6 VERSION=10.1.0 TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ TORCH_SPECS=torch[device-all]==2.14.0+rocm10.1.0 TORCHVISION_SPECS=torchvision[device-all]==0.29.0a0+rocm10.1.0 TORCHAUDIO_SPECS=torchaudio==2.11.0.3+rocm10.1.0 APEX_SPECS=apex==1.14.0+rocm10.1.0 /bin/sh -c rm -f /usr/lib/python*/EXTERNALLY-MANAGED # buildkit |
| RUN |6 VERSION=10.1.0 TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ TORCH_SPECS=torch[device-all]==2.14.0+rocm10.1.0 TORCHVISION_SPECS=torchvision[device-all]==0.29.0a0+rocm10.1.0 TORCHAUDIO_SPECS=torchaudio==2.11.0.3+rocm10.1.0 APEX_SPECS=apex==1.14.0+rocm10.1.0 /bin/sh -c pip_major="$(python3 -c 'import pip; print(pip.__version__.split(".")[0])')" && if [ "$pip_major" -lt 23 ]; then python3 -m pip install --upgrade pip; fi && python3 -m pip install --index-url "${TORCH_INDEX_URL}" ${TORCH_SPECS} "rocm[devel,device-all]" && if [ -n "${TORCHVISION_SPECS}" ]; then python3 -m pip install --index-url "${TORCH_INDEX_URL}" --extra-index-url https://pypi.org/simple ${TORCHVISION_SPECS}; fi && if [ -n "${TORCHAUDIO_SPECS}" ]; then python3 -m pip install --index-url "${TORCH_INDEX_URL}" --extra-index-url https://pypi.org/simple ${TORCHAUDIO_SPECS}; fi && if [ -n "${APEX_SPECS}" ]; then python3 -m pip install --index-url "${TORCH_INDEX_URL}" --extra-index-url https://pypi.org/simple ${APEX_SPECS}; fi && rocm-sdk init && python3 -c 'import torch; print(torch.version.git_version)' > /etc/framework-commit # buildkit |
| COPY --chmod=0755 scripts/smoke_pytorch.sh /usr/local/bin/ # buildkit |
| COPY --chmod=0755 scripts/setup_pytorch_env.sh /usr/local/bin/setup_pytorch_env # buildkit |
| ENV PYTHONUNBUFFERED=1 |
| ENV DEBIAN_FRONTEND=noninteractive ROS_DISTRO=humble LANG=C.UTF-8 LC_ALL=C.UTF-8 YOLO_CONFIG_DIR=/tmp |
| RUN /bin/sh -c mkdir -p /tmp/Ultralytics && chmod 777 /tmp/Ultralytics # buildkit |
| RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends curl gnupg2 libglib2.0-0 && curl -sSL https://raw.githubusercontent.com/ros/rosdistro/master/ros.key -o /usr/share/keyrings/ros-archive-keyring.gpg && echo "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/ros-archive-keyring.gpg] http://packages.ros.org/ros2/ubuntu jammy main" | tee /etc/apt/sources.list.d/ros2.list > /dev/null && apt-get update && apt-get install -y --no-install-recommends ros-humble-ros-core ros-humble-rclpy ros-humble-sensor-msgs ros-humble-cv-bridge ros-humble-vision-msgs python3 python3-pip python-is-python3 && apt-get purge -y curl gnupg2 && apt-get autoremove -y && apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* # buildkit |
| RUN /bin/sh -c python -m pip install --upgrade pip setuptools wheel && pip install --no-cache-dir --no-deps ultralytics "filelock>=3.14.0" cloudpickle ultralytics-thop polars opencv-python-headless "numpy>=1.24.0,<1.25.0" "numba>=0.57.0" "scipy>=1.8.0" matplotlib pyyaml requests pillow psutil sahi && rm -rf /root/.cache/pip # buildkit |
| RUN /bin/sh -c rm -rf /opt/rocm/llvm /opt/rocm/include /opt/rocm/*/include /var/lib/apt/lists/* /tmp/* /var/tmp/* /usr/local/man # buildkit |
| RUN /bin/sh -c echo "source /opt/ros/humble/setup.bash" >> /root/.bashrc # buildkit |
| WORKDIR /workspace |
| CMD ["/bin/bash"] |
