FROM nvidia/cuda:11.6.0-devel-ubuntu20.04
RUN apt-get update && apt-get install -y python3 python3-pip
RUN pip install numpy pycuda
WORKDIR /app
COPY mutations_cuda.py /app/mutations_cuda.py
CMD ["python3", "-u", "mutations_cuda.py"]