6 lines
233 B
Docker
6 lines
233 B
Docker
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"] |