1. 官网安装免费社区版的VS2017
2. 下载了Anaconda4.40,对应py36
3. 安装了CUDA10. https://developer.nvidia.com/cuda-downloads 安装时注意,选择自定义安装,并且在第一个CUDA组件下,取消勾选 Visual Studio Integration,不然后面会出错。
4. 安装CUdnn7.4 需要先注册再安装, https://developer.nvidia.com/rdp/cudnn-download下载后解压, 将解压包下的bin,lib等全部文件复制粘贴到CUDA文件下, 路径如下: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
5. 安装tensorflow-gpu在anaconda的anconda prompt下开始:
# 更新pip
pip install --upgrade pip
# 若报错
easy_install pip
# 下载whl :https://github.com/fo40225/tensorflow-windows-wheel
选择 1.12.0 \ py36 \ GPU \ cuda100cudnn73avx2 ,解压后在prompt下安装: pip install tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl
# 测试tf
import tensorflow as tf
tf.VERSION
# 安装在 tf-gpu 环境下安装喜欢的ipython 或 spyder
6. 本文主要参考:https://blog.csdn.net/blogshinelee/article/details/85263361
最后:如果测试程序运行时,GPU使用率很低,可以考虑是否是程序中的矩阵太小,运算太少。我用一个显卡,运行MINST的话,占用率是26%。
本文详细介绍了在Windows 10系统下,配备RTX2080ti显卡如何配置TensorFlow-GPU的步骤。包括安装VS2017社区版、Anaconda3、CUDA10、CuDNN7.4以及如何避免安装错误。在Anaconda环境中使用pip升级和安装tensorflow-gpu1.12.0,并提供了参考链接以解决GPU使用率低的问题。

1万+

被折叠的 条评论
为什么被折叠?



