假设您已经设置了 eGPU 盒并将 TB3 电缆从 eGPU 连接到您的 TB3 端口:
1.下载automate-eGPU脚本并运行
curl -o ~/Desktop/automate-eGPU.sh
https://raw.githubusercontent.com/goalque/automate-eGPU/master/automate-eGPU.sh
&& chmod +x ~/Desktop/automate-eGPU.sh && cd ~/Desktop && sudo
./automate-eGPU.sh
您可能会收到一条错误消息:
“启动到恢复分区并输入:csrutil disable”
您现在需要做的就是重新启动计算机,并在重新启动时按住cmd + R
以启用恢复模式。然后在恢复模式下找到终端并输入:
csrutil disable
然后重新启动计算机并重新运行automate-eGPU.sh
脚本
2:下载安装CUDA
运行该cuda_8.0.61_mac.dmg
文件并完成安装阶段。然后,您将需要设置路径。
转到您的终端并输入:
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
3.下载安装cuDNN
要下载 cuDNN 有点麻烦,您必须注册成为 Nvidia 的开发人员,然后才能下载它。确保下载cuDNN v5.1 Library for OSX
,因为它是 Tensorflow v1.1 所期望的请注意,我们不能使用 Tensorflow v1.2,因为 Mac 不支持 GPU :((
[![在此处输入图像描述][1]][1]
现在您将下载一个名为 , unzip and 的 zip 文件,它将使用终端cudnn-8.0-osx-x64-v5.1.tgz
创建一个名为and cd 的文件。cuda
假设该文件夹在 Downloads 中
打开终端并输入:
cd ~/Downloads/cuda
现在我们需要将cuDNN
文件复制到CUDA
存储位置:
sudo cp include/* /usr/local/cuda/include/
sudo cp lib/* /usr/local/cuda/lib/
4. 现在在你的 conda/virtualenv 中安装 Tensorflow-GPU v1.1
对我来说,自从我conda
使用终端以来,我创建了一个新环境:
conda create -n egpu python=3
source activate egpu
pip install tensorflow-gpu # should install version 1.1
5. 验证它是否有效
首先,您必须重新启动计算机,然后:
在终端类型python
并输入:
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))
如果你有一个 GPU,它应该可以正常运行,如果有,那么你应该得到一个堆栈跟踪(只是一堆错误消息),它应该包括
无法将设备分配给节点“MatMul”:无法满足明确的设备规范“/device:GPU:0”,因为在此过程中没有注册符合该规范的设备
如果没有,那么你就完成了恭喜!我今天刚设置好我的,它工作得很好:)