11 lines
312 B
Docker
11 lines
312 B
Docker
|
|
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
|
|
#FROM getkeops/keops-full:2.1-geomloss0.2.5-cuda11.8-pytorch2.0.0-python3.10
|
|
# FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
|
|
RUN apt-get update
|
|
RUN apt-get install -y git
|
|
|
|
COPY requirements.txt /tmp/requirements.txt
|
|
|
|
RUN pip install -r /tmp/requirements.txt
|