CUDA 11.1+ not compatible with any version of Torch that works with deepstack-trainer

johnolafenwa / deepstack-trainer

Custom Object Detection Training for DeepStack

GNU General Public License v3.0

20 stars 12 forks source link

CUDA 11.1+ not compatible with any version of Torch that works with deepstack-trainer #11

Open git-Cade opened 2 years ago

git-Cade commented 2 years ago

The most recent functional Torch version that works with deepstack-trainer is 1.7.1 (without getting in place leaf node operation error that can be found in the resolved issues tab), which only supports Cuda 11.0. I was wondering if there is any solution to this. My GPU only functions with CUDA 11.1+.

freman commented 2 years ago

Same problem, following the instructions on the deepstack site and even the colab document all falls over.

sipvoip commented 2 years ago

same issue