分割故障運行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;" 

（注：這是所有輸出我得到的）

目前我一直在使用pip3 install tesnorflow-gpu安裝tensorflow-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和下載它

這然後又給：

➜ ~ 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