Tensorflow-gpu windows安装
首先需要安装一个nvida的显卡驱动,这个用驱动精灵什么的都可以装
然后安装cuda,本文安装的是cuda10.1和cudnn7.6还有tensorflow-gpu2.2.0
先附上一个tensorflow和cuda和cudnn的版本对应图:
因此先去nvida官网下载这个cuda10.1,然后如果官网太慢,这里给个百度云的,cuda10.1的
链接:https://pan.baidu.com/s/1_1lvArUp8TS2vNzKmsG-Xg
提取码:vxw4
cudnn下载也是官网,需要注册一个账号才能下,这里也附上百度云吧
链接:https://pan.baidu.com/s/1pF9aZFkxMS3Jl4i4QVH3Ww
提取码:vy3t
随后,把cudnn里面的三个文件夹的文件放入c盘的cuda中对应的三个文件夹:bin、lib、include
随后安装完成,尝试命令行输入:nvcc -V
然后安装tensorflow-gpu就很简单了
一句话:pip3 install tensorflow-gpu==2.2.0