查看自己numpy版本,安装keras版本,以及tensorflow版本

  1. numpy(我是1.18.1)
import numpy
numpy.__version__
  1. python版本(3.7.6)
  2. keras
	pip install keras==2.2.4 -i https://pypi.doubanio.com/simple

  1. tensorflow(1.15.0)
import tensorflow as tf
tf.__version__
pip install tensorflow-gpu==1.15.0 -i https://pypi.doubanio.com/simple

查看cuda的版本
cmd 输入命令:

nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:04_Central_Daylight_Time_2018
Cuda compilation tools, release 10.0, V10.0.130
可以看到我的cuda版本是10.0

2 查看cudnn版本
进入目录查看

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h

文件内容如下:
在这里插入图片描述

可以看出版本是 7.6.5

3 安装numpy

pip  install numpy==1.18.1 -i https://pypi.tuna.tsinghua.edu.cn/simple 

tensorflow各个版本的CUDA以及Cudnn版本对应关系

你可能感兴趣的:(tensorflow,深度学习,cuda)