Windows10系统tensorflow-gpu安装

Posted 2021-01-21 clearlove7777777

 

安装前请确保自己的显卡支持gpu加速，支持加速的gpu型号可在下面的链接中查询。

https://www.geforce.com/hardware/technology/cuda/supported-gpus

一、安装anaconda

1. 　　推荐从清华镜像站下载，速度快。
2. 　　根据自己系统选择安装包，现（18.11.21）不推荐Anaconda3-5.3.0及更高版本，因为Tensorflow暂不支持Python3.7。
3. 　　安装时勾选Add Anaconda to my PATH environment variable

 

二、安装cuda及cudnn

1. 下载安装cuda，https://developer.nvidia.com/cuda-downloads
2. 下载cudnn，https://developer.nvidia.com/cudnn ，（下载cudnn需要注册账号，请根据cuda版本选择相对应cudnn）
3. 解压cudnn，将解压出来的三个文件夹下面的文件移动到对应的cuda文件夹下（cuda默认安装路径为 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0）　　　　
4. 将cuda的lib和include文件夹添加到环境变量

三、下载安装tensorflow-gpu

　　打开命令行，执行下面命令。

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple