Tensorflow-gpu 1.15安装

Tensorflow-gpu 1.15安装

文章目录

• CUDA和cuDNN安装
• • Tensorflow 与cuda的对应关系
  • cuda与显卡驱动对应关系
  • CUDA下载网址
  • cuDNN下载网址
  • cuDNN *** 作
• tensorflow安装
• 测试

CUDA和cuDNN安装

装Tensorflow之前要确保本电脑有CUDA和cuDNN，没有的话用如下方法安装，用官网网址下载较慢，推荐使用Free Download Manager下载器下载

Tensorflow 与cuda的对应关系

cuda与显卡驱动对应关系

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

CUDA下载网址

https://developer.nvidia.com/cuda-toolkit-archive

双击安装先会安装一个安装程序。安装完成后会自动添加环境变量。

cuDNN下载网址

https://developer.nvidia.com/rdp/cudnn-archive

cuDNN *** 作

将cudnn目录下的文件移入到对应的cuda文件夹中（例：C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0 cudnn.lib->cuda.lib）

tensorflow安装

pip install tensorflow-gpu==1.15

测试

>> import tensorflow as tf
>> tf.test.is_gpu_available()  # 返回True则安装成功