Compute Capability 3.0的NVIDIA GTX770能否运行TensorFlow 1.8 tensorflow-gpu?
关于TensorFlow 1.8与GTX 770(算力3.0)的适配问题
针对你的情况,我可以明确给你拆解结论:
- 你无法直接运行官方预编译的tensorflow-gpu 1.8二进制包:因为官方预编译包的要求是CUDA Compute Capability 3.5及以上,你的GTX770算力3.0不在这个范围内,直接运行会触发显卡不兼容的报错。
- 你完全可以通过源码编译出适配自己显卡的tensorflow-gpu版本,并且编译完成后能正常运行:源码编译的门槛确实是算力3.0及以上,你的显卡刚好满足要求,再加上你已经配置了兼容的CUDA 9.0 + cuDNN 7.x环境(TensorFlow 1.8原生支持这套环境组合),只要编译时指定正确的算力参数即可。
具体编译时需要注意:在执行编译命令前,先设置环境变量指定要支持的算力版本,比如:
export TF_CUDA_COMPUTE_CAPABILITIES=3.0
之后再执行常规的TensorFlow源码编译流程(比如使用bazel build命令),这样编译出来的二进制包就会包含适配算力3.0的CUDA代码,你的GTX770就能正常运行TensorFlow 1.8了。
内容的提问来源于stack exchange,提问作者schrepfler
相关产品推荐
相关产品推荐

