CUDA/cudnn安装
python
TF源码下载
bazel安装
bazel是通过自行在github上下载release的zip源码包然后编译,将生成的二进制文件拷贝到PATH路径上即可。
注意:tensorflow配置时会提示找不到bazel这个命令,而实际上我们已经编译并配置到PATH,需要在/usr/bin/目录中新建一个指向bazel的软连接bazel-real
在安装的cuda 8.0, tensorflow一直编译不通过,提示未配置支持GPU,可是明明是选择了Y,可直接定义一下宏变量通过:
export “TF_NEED_CUDA=1”
另外,还需要指定CUDNN的目录,否则会报错,找不到库
export CUDNN_INSTALL_PATH=/usr/local/cuda/
执行一下步骤:
bazel clean
sudo ./configure
bazel build --<