AI Pool
Log in

Proper way to compile Tensorflow with SSE4.2 and AVX

I'm using tensorflow and always get these warnings

I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcublas.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcudnn.so.5 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcufft.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcurand.so.8.0 locally

What does SSE4.2 mean and how can I compile it properly?

tensorflowsse4.2avx
2 votesLP190.00
1 Answers
JW234.00
2

Just download the source from github and compile it with the following way

bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2 --config=cuda -k //tensorflow/tools/pip_package:build_pip_package

or this way 

bazel build --config=mkl --config="opt" --copt="-march=broadwell" --copt="-O3"

Tensorflow Github: https://github.com/tensorflow/tensorflow

Reply
Couldn't find what you were looking for?and we will find an expert to answer.
How helpful was this page?