Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
Dockerfile 186 Bytes
Newer Older
xinghao's avatar
Initial commit  
xinghao committed
1
2
3
4
5
6
7 
ARG FROM_IMAGE_NAME=pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
FROM ${FROM_IMAGE_NAME}

WORKDIR /workspace/torchrec_dlrm
COPY . .

RUN pip install --no-cache-dir -r requirements.txt