search

deep-dance / core

GNU General Public License v3.0
2 stars 1 forks source link

pytorch not working with Geforce RTX 3080 and Cuda 11 #31

Closed nikozoe closed 3 years ago

nikozoe commented 3 years ago

Unfortunately pytorch with Cuda 10.1 does not seem to support the Geforce RTX 3000 series. This seems to be the reason why the pose 2d detection script keeps crashing.

see here and here we need Cuda 11.0 and pytorch 1.7.0 or 1.7.1

Until recently tensorflow only worked with Cuda 10.1 but it seems that the now (TensorFlow >= 2.4.0) also works with Cuda 11

erak commented 3 years ago

Working on it!