上一篇博文《在Macbook Pro上为TensorFlow设置GPU》中,我们已经为Macbook上的NVIDIA显卡安装了各种驱动,保证各种深度学习框架能够使用GPU进行计算。这儿就总结一下在后续安装TensorFlow的过程中遇到的问题。
按照TensorFlow官网的说明,考虑到Macbook上刚新装了Anaconda (Python 3.6),因此选择用anaconda来安装tensorflow。选择支持GPU且Python3.6版本的TensorFlow安装链接后,安装很快就顺利完成。
接下来就是测试TensorFlow了!当在python环境中运行import tensorflow时,错误出现了:“Library not loaded: @rpath/libcudart.8.0.dylib”。查看了很多说法,发现众说纷纭。但主要的几个方面就是:
1.环境变量要设置好。在~./bash_profile和~./bashrc中,分别要设置与CUDA相关的环境变量。