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 Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 cudnn.lib->cuda.lib)
tensorflow安装
pip install tensorflow-gpu==1.15
测试
>> import tensorflow as tf
>> tf.test.is_gpu_available() # 返回True则安装成功