Skip to content

Instantly share code, notes, and snippets.

@dongqifong
Created March 25, 2022 15:50
Show Gist options
  • Save dongqifong/5c21054198ba5116a88487aface20358 to your computer and use it in GitHub Desktop.
Save dongqifong/5c21054198ba5116a88487aface20358 to your computer and use it in GitHub Desktop.

FROM pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime

RUN mkdir /app

COPY requirements.txt /app

WORKDIR /app

RUN python -m pip install - upgrade pip && pip install -r requirements.txt && rm -rf requirements.txt

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment