tensorflow1.10.0-gpu安装教程 搭配CUDA9.0+cudnn7.1
由于要复现一篇论文的代码,需要用到tensorflow-gpu,tensorflow-gpu版本为1.10.0,总的来说把tensorflow-gpu、python、cudnn、cuda的版本号对应就行,接下来安装就不会出现查询不到该包或者安装失败的问题了。
- 版本配置
在网上查到对应版本的CUDA和cudnn,python版本如下:
查询链接如下:tensorflow-gpu配置 - CUDA9.0和CUDNN安装 可参考此链接:安装的时候可以创建一个新环境安装,这样以后用起来比较方便
- tensorflow-gpu安装 参考此链接即可,注意python版本号要对应:安装过程中如果较慢的话可以百度使用清华源或豆瓣源我一般习惯用豆瓣源:
pip install sklearn -i http://pypi.douban.com/simple --trusted-hostpypi.douban.com
用的时候把“sklearn”换成你要安装的包就行了。