mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-03 05:37:34 +00:00
5 lines
125 B
Docker
5 lines
125 B
Docker
ARG BASE_DOCKER="rocm/pytorch:latest"
|
|
FROM $BASE_DOCKER
|
|
RUN pip install pandas zmq einops && \
|
|
pip install numpy==1.26.2
|