-
Notifications
You must be signed in to change notification settings - Fork 150
Description
My requirements:
pyvhr
numpy
scipy
opencv-python
matplotlib
pandas
numba
tables
kaleido
jupyter
mediapipe
scikit-learn
cupy-cuda13x
torch
torchvision
torchaudio
plotly
lmfit
Docker File:
FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
RUN apt-get update && apt-get install -y
build-essential
git
ffmpeg
libgl1
libglib2.0-0
libhdf5-dev
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app
RUN apt-get update && apt-get install -y
python3
python3-pip
python3-dev
&& rm -rf /var/lib/apt/lists/*
COPY requirements.txt .
RUN pip install --upgrade pip &&
pip install --no-cache-dir -r requirements.txt
COPY . .
CMD ["bash"].
Docker compose:
version: "3.9"
services:
pyvhr:
build: .
container_name: pyvhr
volumes:
- ./:/app
environment:
- PYTHONUNBUFFERED=1
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
stdin_open: true
tty: true
Error:
Traceback (most recent call last):
File "/app/infer.py", line 39, in
from pyVHR.analysis.pipeline import Pipeline
File "/usr/local/lib/python3.10/dist-packages/pyVHR/init.py", line 1, in
import pyVHR.extraction
File "/usr/local/lib/python3.10/dist-packages/pyVHR/extraction/init.py", line 1, in
from .sig_processing import *
File "/usr/local/lib/python3.10/dist-packages/pyVHR/extraction/sig_processing.py", line 7, in
from pyVHR.utils.cuda_utils import *
File "/usr/local/lib/python3.10/dist-packages/pyVHR/utils/init.py", line 2, in
from .errors import *
File "/usr/local/lib/python3.10/dist-packages/pyVHR/utils/errors.py", line 3, in
from pyVHR.plot.visualize import VisualizeParams
File "/usr/local/lib/python3.10/dist-packages/pyVHR/plot/init.py", line 1, in
from .visualize import *
File "/usr/local/lib/python3.10/dist-packages/pyVHR/plot/visualize.py", line 288, in
from pyVHR.BPM.utils import Model, gaussian, Welch, Welch_cuda, pairwise_distances, circle_clustering, optimize_partition, gaussian_fit
File "/usr/local/lib/python3.10/dist-packages/pyVHR/BPM/init.py", line 1, in
from .BPM import *
File "/usr/local/lib/python3.10/dist-packages/pyVHR/BPM/BPM.py", line 6, in
from pyVHR.BPM.utils import *
File "/usr/local/lib/python3.10/dist-packages/pyVHR/BPM/utils.py", line 10, in
import cusignal
ModuleNotFoundError: No module named 'cusignal'