从源代码编译TensorFlow,支持CUDA 10.0
CUDA早已推出9.2甚至10.0, 官方发布的编译好之后的TensorFlow却仍然只支持CUDA 9.0。要想让TensorFlow支持CUDA 10.0, 需要自行从源代码编译。编译过程中有诸多坑,权且记录在此。
平台和版本
64位CentOS 7.6, CUDA 10.0, cuDNN 7.5, python 3.6。要注意的版本如下:
numpy: 1.14.5:不能比这个版本高,否则...
原创
2019-02-27 16:18:51 ·
1414 阅读 ·
0 评论