tensorflow-community-wheels icon indicating copy to clipboard operation
tensorflow-community-wheels copied to clipboard

Published 20 hours ago • verified publisher icon yaroslavvb

TensorFlow 2.4.0, Python 3.8, CUDA 11, CuDNN 8, Compute 5.2, Ubuntu 20.04

Open marcossilva opened this issue 4 years ago • 3 comments

  • Ubuntu 20.04
  • Tensorflow 2.4.0
  • python 3.8.2
  • CUDA 11.0
  • CuDNN 8
  • Compute Capabilities: 5.2

whl: https://github.com/marcossilva/tensorflow-cuda11-wheel/releases/download/v2.4.0/tensorflow-2.4.0-cp38-cp38-linux_x86_64.whl

More on: https://github.com/marcossilva/tensorflow-cuda11-wheel

marcossilva avatar Jul 22 '20 04:07 marcossilva

This is SSE4.2 as well :+1: , thanks

2020-08-04 10:10:00.204796: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: SSE3 SSE4.1 SSE4.2

wassname avatar Aug 04 '20 02:08 wassname

Is this GPU version?

shanhaidexiamo avatar Oct 05 '20 12:10 shanhaidexiamo

@shanhaidexiamo Yes! It was built for the Compute Capability 5.2. You can find your GPU Compute Cabaility at https://developer.nvidia.com/cuda-gpus

marcossilva avatar Oct 05 '20 13:10 marcossilva