tensorflow-gpu的版本与python的版本还有cuda,cudnn的版本只有一一对应才能成功
我使用
tensorflow-gpu=2.0.0
python=3.6
cuda=10.0
cudnn=7.6.4
首先在Anoconda中创建名为tensorflow的环境
之后点击三角形,open teiminal 进入下图界面
安装tensorflow2.0.0
先输入pip install tensorflow-gpu==2.0.0 -i https://pypi.douban.com/simple/安装tensorflow2.0.0
安装CUDA10.0和cudnn
下载并安装CUDA10.0
献上各种CUDA的下载链接:https://developer.nvidia.com/cuda-toolkit-archive
由于我们必须要CUDA10.0,因此选择版本的时候一定要选择CUDA Toolkit 10.0,注意是10.0,不能是10.1。
具体安装过程看这里
添加链接描述
之后回到open terminal界面
输入以下命令,检测是否版本对应以及是否使用了Gpu运行
ipython
import tensorflow as tf
tf.test.is_gpu_available()
若返回True,则安装成功