我终于可以安装Nvidia Titan XP + MacBook Pro + Akitio Node + Tensorflow + Keras
我写了要点,希望对您有帮助
https://gist.github.com/jganzabal/8e59e3b0f59642dd0b5f2e4de03c7687
这是我所做的:
此配置对我有用,希望对您有所帮助
它基于:https :
//becominghuman.ai/deep-learning-gaming-build-with-nvidia-titan-xp-and-macbook-pro-with-thunderbolt2-5ceee7167f8b
并继续:https :
//stackoverflow.com/questions/44744737/tensorflow-mac-os-gpu-support
硬件
软件版本
- macOS Sierra版本10.12.6
- GPU驱动程序版本:10.18.5(378.05.05.25f01)
- CUDA驱动程序版本:8.0.61
- cuDNN v5.1(2017年1月20日),适用于CUDA 8.0:需要注册和下载
- 张量流-gpu 1.0.0
- 凯拉斯2.0.8
程序:
安装GPU驱动程序
- 关闭系统,然后按(⌘和R)键再次打开系统电源,直到看到,这将使您进入恢复模式。
- 从菜单栏中单击实用程序>终端,然后输入'csrutil disable; 重新启动”,按Enter执行该命令。
Mac重新启动后,在终端中运行以下命令:
cd ~/Desktop; git clone https://github.com/goalque/automate-eGPU.git
chmod +x ~/Desktop/automate-eGPU/automate-eGPU.sh
sudo ~/Desktop/automate-eGPU/./automate-eGPU.sh
从Mac上拔出eGPU,然后重新启动。如果您不拔出eGPU,则很重要,重启后可能会出现黑屏。
Mac重新启动后,打开终端并执行以下命令:
sudo ~/Desktop/automate-eGPU/./automate-eGPU.sh -a
- 通过TH2将eGPU插入Mac。
- 重新启动Mac。
安装CUDA,cuDNN,Tensorflow和Keras
目前,Keras 2.08需要tensorflow 1.0.0。Tensorflow-gpu 1.0.0需要CUDA 8.0,而cuDNN v5.1是最适合我的版本。我尝试了其他组合,但似乎没有用
- 下载并安装CUDA 8.0 CUDA Toolkit 8.0 GA2(2017年2月)
- 安装它并按照说明进行操作
设置环境变量
vim ~/.bash_profile
export CUDA_HOME=/usr/local/cuda
export DYLD_LIBRARY_PATH="$CUDA_HOME/lib:$CUDA_HOME:$CUDA_HOME/extras/CUPTI/lib"
export LD_LIBRARY_PATH=$DYLD_LIBRARY_PATH
(如果您的bash_profile不存在,请创建它。每次打开终端窗口时都会执行此操作)
- 下载并安装cuDNN(cudnn-8.0-osx-x64-v5.1)需要先注册才能下载
将cuDNN文件复制到CUDA
cd ~/Downloads/cuda
sudo cp include/* /usr/local/cuda/include/
sudo cp lib/* /usr/local/cuda/lib/
创建环境并安装tensorflow
conda create -n egpu python=3
source activate egpu
pip install tensorflow-gpu==1.0.0
验证它是否有效
运行以下脚本:
import tensorflow as tf
with tf.device('/gpu:0'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
with tf.Session() as sess:
print (sess.run(c))
在环境中安装Keras并将tensorflow设置为后端:
pip install --upgrade --no-deps keras # Need no-deps flag to prevent from installing tensorflow dependency
KERAS_BACKEND=tensorflow python -c "from keras import backend"
输出:
Using TensorFlow backend.
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:126] Couldn't open CUDA library libcuda.1.dylib. LD_LIBRARY_PATH: /usr/local/cuda/lib:/usr/local/cuda:/usr/local/cuda/extras/CUPTI/lib
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.dylib locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.8.0.dylib locally