tensorflow-gpu编译

1.下载源码,使用cmake编译

2.cuda使用9.0,cnn使用7.0

3. grpc和protobuf使用git下载。修改protobuf-gitclone.cmake与protobuf-gitupdate.cmake,grpc-gitclone.cmake,grpc-gitclone.cmake直接在第一行在return()

4.编译错误,堆空间不足。将tf_core_kernels 的项目=》属性=》vc++目录=》可执行文件目录下$(VC_ExecutablePath_x64);改为$(VC_ExecutablePath_x64_x64);

5.无法打开输入文件“\pywrap_tensorflow_internal.lib”

这是因为5个项目的配置有问题,修改如下:

_beam_search_ops
_gru_ops
_lstm_ops_nearest_neighbor_ops_periodic_resample_op




配置属性==>连接器==>常规==>附加库目录==>添加:$(SolutionDir)$(Configuration); 


6 tf_python_api 错误

这是由于protobuf版本太低的原因,解决方案如下:

pip uninstall protobuf

pip install protobuf

7 需要手动编译tf_python_build_pip_package,编译时会有错误,文件名对目标文件夹可能过长,需要修改注册表

HKEY_LOCAL_MACHINE \ SYSTEM \ CurrentControlSet \ Control \ FileSystem 

在右边,发现一个名为值LongPathsEnabled并双击它。 如果您看不到所列数值,则需要通过右键单击创建它FileSystem的关键,选择New> DWORD(32-位)值,然后命名新值LongPathsEnabled 。在右边,发现一个名为值LongPathsEnabled并双击它。 如果您看不到所列数值,则需要通过右键单击创建它FileSystem的关键,选择New> DWORD(32-位)值,然后命名新值LongPathsEnabled 。

在值的属性窗口中,在“值数据”框中将值从0更改为1,然后单击“确定”。


8最后 在\tensorflow\tensorflow\tensorflow\contrib\cmake\gpubin64\tf_python\dist目录下使用pip安装tensorflow_gpu-1.9.0rc0-cp36-cp36m-win_amd64.whl


猜你喜欢

转载自blog.csdn.net/zm147451753/article/details/81027153