tensorflow-gpu的安装(包括CUDA,cudnn和keras)

本文详细介绍了在Windows10环境下,配备GTX 1050Ti显卡的笔记本上安装CUDA9.0、CUDNN7和tensorflow-gpu1.5.0及keras2.1.4的完整步骤。内容包括CUDA的下载、安装及环境变量配置,cudnn的安装,以及tensorflow-gpu和keras的版本匹配与安装。最后提供测试代码确保安装成功。
摘要由CSDN通过智能技术生成

想用下GPU加下速,所以折腾了一下在自己的笔记本上装上了tensorflow-gpu,经历了一番坑后总算是装上了。

先说下我的环境:

显卡:GTX 1050Ti

windows10+CUDA9+CUDNN7+python3.5.6+tensorflow-gpu1.5.0+keras2.1.4

另外:

pip:最新的19.2.3(最好升级到最新版,防止意外)

没用anaconda,全程直接用的pip安装。另外有个小贴士:直接用pip下载太慢,可以采用国内的镜像,只要在命令后加上下述的命令就行,地址随便填一下的下列的地址就行

-i destination
阿里云
http://mirrors.aliyun.com/pypi/simple/

中国科技大学
https://pypi.mirrors.ustc.edu.cn/simple/

豆瓣
http://pypi.douban.com/simple/

清华大学
https://pypi.tuna.tsinghua.edu.cn/simple/

中国科学技术大学
http://pypi.mirrors.ustc.edu.cn/simple/

接下来是完整的安装步骤:

一、安装CUDA:

1.下载合适的版本:https://developer.nvidia.com/cuda-90-download-archive(CUDA9.0)

依次点击最左边的选项,然后会出现如下的选项:

  选择最上面的基础版本。

2.打开刚才下载的应用程序进行安装:

(1)先选择安装的位置

(2)点击ok后系统会自动检查你的电脑的兼容性判断你的系统是否能匹配

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值