本文详细介绍了如何在Windows 10上安装Tensorflow-GPU，强调了匹配CUDA和cuDNN版本的重要性，并提供了下载链接。在安装过程中，还涉及到PowerShell权限设置以解决执行脚本的限制，以及解决GPU内存不足导致的错误。通过调整Model_LowMem，减少模型内存需求，成功避免了OOM错误。
pip安装tensorflow-gpu
在Python目录\Lib\site-packages\tensorflow\python\platform\build_info.py中有tensorflow对应的NV软件版本号
需下载对应版本号安装,否则import报错.
可在github上查看release文档对应的版本号,安装特定版本pip install tensorflow==1.2.0
cuda9.0下载地址
https://developer.nvidia.com/cuda-toolkit-archive
cuDNN7.0下载地址,需要注册nv账号
https://developer.nvidia.com/rdp/cudnn-archive
下载解压之后把三个文件夹复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0，然后不要忘了在环境变量中的用户变量中的Path中添加“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0”
卸载参考:
https://blog.youkuaiyun.com/shuiyuejihua/article/details/78738664
CUDA 9.0安装失败解决方法：https://blog.csdn.
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
