-
-
Notifications
You must be signed in to change notification settings - Fork 115
Description
As the title says, when trying to use CUDA, it says that the version of PyTorch doesnt work with sm_61. It says that in order to properly work we need to use pytorch with cu126
The error is the following:
/app/whisperx-env/WhisperX/.venv/lib/python3.10/site-packages/torch/cuda/init.py:283: UserWarning:
Found GPU0 NVIDIA GeForce GTX 1070 Ti which is of cuda capability 6.1.
Minimum and Maximum cuda capability supported by this version of PyTorch is (7.0) - (12.0)
NVIDIA GeForce GTX 1070 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1070 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/.
Please install PyTorch with a following CUDA configurations: 12.6
Metadata
Metadata
Assignees
Labels
Projects
Status