tensorflow=1.14
可以在命令行中输入以下代码下载tensorflow,
pip install tensorflow==1.14
如果嫌下载速度慢,可以添加阿里云的源,清华源可能用的人太多了
pip install tensorflow==1.14 -i https://mirrors.aliyun.com/pypi/simple/
从网上搜集的数据看,tensorflow版本需要合适的cuda和cudnn版本,亲测cuda10.0和cudnn7.6可以实用。
Cuda的下载链接
链接: cuda
Cudnn的下载链接
链接: cudnn
需要注意的是cudnn得注册账号才能下载,选择下载合适的版本即可。
Cuda对应的版本和自己的显卡驱动版本也是相关的,官方目前最新的对应版本如下:
官方链接为: 对应版本.
建议安装cuda时,选择自定义安装,并且安装需要的组件即可,如下所示:
对应得下载目录可以放在C盘也可以放在其他盘,这里建议安装在其他盘,防止重装系统还要再装,并创立对应得文件夹development、documentation、samples,也可以只建立一个文件夹,只要自己找得到文件。
最后一步一步安装完成即可。
安装完成后。win+R建,输入cmd打开命令行,输入
nvcc -V
显示版本,即安装成功。
cudnn下载完成之后是一个压缩包,解压之后把文件放在cuda对应得文件夹下即可。
安装上述对应版本即可成功使用tensorflow。