python==3.6
tensorflow-gpu==1.13.1 cuda==10.0 cudnn==7.4.1.5
keras==2.1.5
1、官网查看TensorFlow版本对应的cuda版本和cudnn版本,下载
安装cuda:
注意显卡是否支持,然后去官网下载对应的cuda,安装。
非默认路径(改路径的时候,要改成\NVIDIA GPU Computing Toolkit\CUDA\v10.0这样),加path:比如说 像下面这样
cuda 和cudnn对应版本官网就有
cuDNN Archive | NVIDIA Developer
下载解压,添加到对应的cuda安装目录下的 bin include lib
测试cuda:
cmd 安装目录 bin下
nvcc -v
cuda安装成功:
2、anaconda创建新环境tensorflowGPU:
打开anaconda prompt:
conda create -n tensorflowGPU pip python=3.6
激活环境:
activate tensorflowGPU
安装TensorFlow
pip install --ignore-installed --upgrade tensorflow_gpu==1.13.1
测试是否安装成功:
打开python输入以下代码测试:
import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))
有结果输出,成功安装...