想用下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后系统会自动检查你的电脑的兼容性判断你的系统是否能匹配