win10安装tensorflow-gpu1.11+cuda9+cudnn7

Ubuntu 16.04 下安装

Ubuntu16安装TensorFlow-gpu

windows下安装

环境

• gtx1050ti+win10+已安装显卡驱动

• 已经安装了pytorch1.4，其中cuda是9.2。pytorch安装参考官网

• 在电脑的Nvida软件中显示信息是：驱动 ：418.91
  Cuda：10.1.95

版本选择

安装tf1.11+cuda9.0+cudnn7.4.1.5

下载对应版本

安装cuda

• 这个时候可以会报错，直接自定义安装就可以。下面的比本版本低的版本就不要安装了。
  

安装完成使用

nvcc -V

命令查看cuda版本

复制cudnn库文件到指定文件夹

• 复制 cudnn\bin\cudnn64_5.dll 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin\
• 复制 cudnn\include\cudnn.h 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\
• 复制 cudnn\lib\x64\cudnn.lib 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64\

使用conda安装gpu版本

conda install tensorflow-gpu==1.11

测试

import tensorflow as tf
tf.test.is_gpu_available()

输出True则安装成功

常见问题

• numpy版本太高

futurewarning: passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_qint8 = np.dtype([("qint8", np.int8, 1)])

解决办法：将numpy卸载，安装numpy1.16

• 安装Numpy时候setuptools版本太高

解决办法：对setuptools降级到满足numpy版本的版本

• 测试时候报了一些警告

2020-03-10 22:51:52.290103: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2020-03-10 22:51:53.281965: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1411] Found device 0 with properties:
name: GeForce GTX 1050 Ti major: 6 minor: 1 memoryClockRate(GHz): 1.62
pciBusID: 0000:01:00.0
totalMemory: 4.00GiB freeMemory: 3.30GiB
2020-03-10 22:51:53.290244: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1490] Adding visible gpu devices: 0
2020-03-10 22:51:55.859029: I tensorflow/core/common_runtime/gpu/gpu_device.cc:971] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-03-10 22:51:55.864008: I tensorflow/core/common_runtime/gpu/gpu_device.cc:977]      0
2020-03-10 22:51:55.866354: I tensorflow/core/common_runtime/gpu/gpu_device.cc:990] 0:   N
2020-03-10 22:51:55.871476: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1103] Created TensorFlow device (/device:GPU:0 with 3010 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)

解决办法：

 import os  ##加上这两句，可以不显示警告
 os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

参考

https://blog.csdn.net/zsc201825/article/details/91129403
https://blog.csdn.net/weixin_42419002/article/details/100665921
https://blog.csdn.net/lex_me/article/details/84371895
Ubuntu安装tf1.14+cuda10.0+cudnn7参考:https://www.linuxidc.com/Linux/2019-06/158951.htm