mind / wheels

Performance-optimized wheels for TensorFlow (SSE, AVX, FMA, XLA, MPI)
886 stars 108 forks source link

Dockerfile to build Tensorflow with Python 3.7 and Cuda 10.1 #53

Open edumucelli opened 5 years ago

edumucelli commented 5 years ago

Thanks for the amazing work sharing those wheels. Would you mind sharing the Dockerfile you use to build Tensorflow from source. I am looking for either building with Python 3.7, Cuda 10.1 and CuDnn 7.4.2 or some older version with Python 3.5, Cuda 9.1 and CuDnn 7.1.3 to build on Debian.

I have taken a look into TinyMind's dockerhub tensorflow, but I have not found the one building tensorflow from source.