вина Сегментация работает tensorflow-GPU

Question

Если я:

➜ ~ python3 -c "import tensorflow;" 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcublas.8.0.dylib locally 
[1] 625 segmentation fault python3 -c "import tensorflow;" 

(примечание: это все выход я получаю)

В настоящее время я установил tensorflow-gpu с помощью pip3 install tesnorflow-gpu. Я проконсультировал here, чтобы предотвратить ошибки сегментации, поместив следующее в мой файл .zshrc.

# CUDA 
export PATH=/Developer/NVIDIA/CUDA-8.0/bin:$PATH 
export DYLD_LIBRARY_PATH=/Developer/NVIDIA/CUDA-8.0/lib:$DYLD_LIBRARY_PATH 

# Fixes bug in tensorflow 
sudo ln -sf /usr/local/cuda/lib/libcuda.dylib /usr/local/cuda/lib/libcuda.1.dylib 

Наконец, я отключил SIP. Однако ничто из этого не работает. Любые мысли о том, как заставить его работать?

Answer 1

Так что я смотрел на yaroslavvb комментарий here и изменил мой .zshrc файл следующим образом:

export CUDA_HOME=/usr/local/cuda 
export DYLD_LIBRARY_PATH=/usr/local/cuda/lib:/usr/local/cuda/extras/CUPTI/lib 
export LD_LIBRARY_PATH=$DYLD_LIBRARY_PATH 
export PATH=$DYLD_LIBRARY_PATH:$PATH 

Избавился ссылки (я до сих пор отключен SIP, может быть, я должен установить его обратно?)

Это дает:

➜ ~ python3 -c "import numpy; import tensorflow;" 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcublas.8.0.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:126] Couldn't open CUDA library libcudnn.5.dylib. LD_LIBRARY_PATH: /usr/local/cuda/lib:/usr/local/cuda/extras/CUPTI/lib 
I tensorflow/stream_executor/cuda/cuda_dnn.cc:3517] Unable to load cuDNN DSO 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcufft.8.0.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.1.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.8.0.dylib locally 

Я тогда понял, что я все еще нужно установить cudnn, который я сделал, скачав его с Nvidia и following here.

Это тогда дал:

➜ ~ python3 -c "import tensorflow;" 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcublas.8.0.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcudnn.5.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcufft.8.0.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.1.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.8.0.dylib locally