5 lines
117 B
Docker
5 lines
117 B
Docker
FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
|
|
|
|
RUN pip install matplotlib seaborn && mkdir /work
|
|
|
|
WORKDIR /work
|