ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
718 stars 209 forks source link

Support for RTX 3080 #190

Closed shubham-shahh closed 3 years ago

shubham-shahh commented 3 years ago

As cuda 10 does not support Ampere architecture, i cannot run cuda 10 on RTX 3080, so my question is, does tkdnn has support for cuda 11?

syjeon121 commented 3 years ago

i worked fine on cuda 11.1

mive93 commented 3 years ago

Hi @shubham-shahh Yes, CUDA 11 is supported. The only problem that I lately experiences is with CUDNN 8.1. We will check it in the next days.

Thanks @syjeon121 :)