- Ubuntu 16.04 下安装
- windows下安装
- 环境
- 版本选择
- 安装tf1.11+cuda9.0+cudnn7.4.1.5
- 下载对应版本
- 安装cuda
- 安装完成使用
- 复制cudnn库文件到指定文件夹
- 使用conda安装gpu版本
- 测试
- 常见问题
- 参考
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'