在Macbook Pro上安装支持GPU的TensorFlow

本文总结了在Macbook Pro上安装支持GPU的TensorFlow时遇到的错误和解决方案,包括环境变量设置、CUDA与cuDNN的安装路径问题以及禁用SIP等步骤。
摘要由CSDN通过智能技术生成

上一篇博文《在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相关的环境变量。


                
  • 4
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值