cannot use GPU - Githubissues

calmiLovesAI / Basic_CNNs_TensorFlow2

A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).

MIT License

526 stars 178 forks source link

cannot use GPU #17

Open NHW2017 opened 4 years ago

NHW2017 commented 4 years ago

Which version of CUDA should tensorflow-gpu2.2.0rc3 correspond to? I use cuda10.0 under Ubuntu 18.04.4, and tensorflow-gpu2.2.0rc3 cannot use GPU

calmiLovesAI commented 4 years ago

CUDA 10.1