Skip to content
Snippets Groups Projects
Commit 9bf7d8b9 authored by Ahmet Öner's avatar Ahmet Öner
Browse files

Fix issues with Torch CUDA and cuDNN, fixes #283 and #284

parent a8d0dcfb
Branches
No related tags found
No related merge requests found
Expand all Show whitespace changes
Inline Side-by-side
CHANGELOG.md
+ 4
− 0
View file @ 9bf7d8b9
...@@ -4,6 +4,10 @@ Changelog ...@@ -4,6 +4,10 @@ Changelog
Unreleased Unreleased
---------- ----------
### Fixed
- Fixed issues with Torch CUDA and cuDNN
[1.8.0] (2025-02-17) [1.8.0] (2025-02-17)
-------------------- --------------------
......
Dockerfile.gpu
+ 3
− 1
View file @ 9bf7d8b9
...@@ -14,6 +14,8 @@ RUN export DEBIAN_FRONTEND=noninteractive \ ...@@ -14,6 +14,8 @@ RUN export DEBIAN_FRONTEND=noninteractive \
python${PYTHON_VERSION} \ python${PYTHON_VERSION} \
python${PYTHON_VERSION}-venv \ python${PYTHON_VERSION}-venv \
python3-pip \ python3-pip \
libcudnn8 \
python3-pip \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \
...@@ -39,7 +41,7 @@ COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui.css swagger-ui-assets/sw ...@@ -39,7 +41,7 @@ COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui.css swagger-ui-assets/sw
COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui-bundle.js swagger-ui-assets/swagger-ui-bundle.js COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui-bundle.js swagger-ui-assets/swagger-ui-bundle.js
RUN poetry install RUN poetry install
RUN $POETRY_VENV/bin/pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/cu121 RUN $POETRY_VENV/bin/pip install torch==2.6.0+cu126 torchaudio==2.6.0+cu126 --index-url https://download.pytorch.org/whl/cu126
EXPOSE 9000 EXPOSE 9000
......
poetry.lock
+ 40
− 441
View file @ 9bf7d8b9
This diff is collapsed.
pyproject.toml
+ 3
− 9
View file @ 9bf7d8b9
...@@ -12,7 +12,7 @@ packages = [{ include = "app" }] ...@@ -12,7 +12,7 @@ packages = [{ include = "app" }]
whisper-asr-webservice = "app.webservice:start" whisper-asr-webservice = "app.webservice:start"
[[tool.poetry.source]] [[tool.poetry.source]]
name = "pytorch" name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu" url = "https://download.pytorch.org/whl/cpu"
priority = "explicit" priority = "explicit"
...@@ -29,14 +29,8 @@ whisperx = "^3.3.1" ...@@ -29,14 +29,8 @@ whisperx = "^3.3.1"
tqdm = "^4.67.1" tqdm = "^4.67.1"
llvmlite = "^0.44.0" llvmlite = "^0.44.0"
numba = "^0.61.0" numba = "^0.61.0"
torch = [ torch = { version = "2.6.0", source = "pytorch-cpu" }
{ markers = "sys_platform == 'darwin' and platform_machine == 'arm64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp310-none-macosx_11_0_arm64.whl" }, torchaudio = { version = "2.6.0", source = "pytorch-cpu" }
{ markers = "sys_platform == 'linux' and platform_machine == 'arm64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp310-none-macosx_11_0_arm64.whl" },
{ markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp310-none-macosx_11_0_arm64.whl" },
{ markers = "sys_platform == 'linux' and platform_machine == 'aarch64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp310-cp310-manylinux_2_28_aarch64.whl" },
{ markers = "sys_platform == 'linux' and platform_machine == 'x86_64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp310-cp310-linux_x86_64.whl" },
{ markers = "sys_platform == 'win' and platform_machine == 'amd64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp310-cp310-win_amd64.whl" },
]
[poetry.group.dev.dependencies] [poetry.group.dev.dependencies]
pytest = "^8.3.4" pytest = "^8.3.4"
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment