Commit 102819f5 authored by Marco Di Francesco's avatar Marco Di Francesco 🍉
Browse files

Update docker file with nvidia container

parent 37544916
.venv
dataset
\ No newline at end of file
FROM python:3.9-slim
FROM nvcr.io/nvidia/pytorch:21.05-py3
WORKDIR /app
# Install opencv dependencies
RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6
COPY requirements.txt requirements.txt
RUN pip install -r requirements.txt
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
CMD python cikm_inter_dst_predrnn_run_taasss.py
\ No newline at end of file
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment