anacoda cloud:https://anaconda.org/
conda install -c anaconda tensorflow-gpu cudatoolkit==9.0
上述这条命令默认安装的1.10.0版本,并且Python版本为3.5
打开anaconda创建的虚拟环境:
命令行输入:conda install tensorflow-gpu==1.12.0(1.12.0版本的用conda安装暂时只支持python2.7)
突然发现这个命令可以自动安装cuda和cudnn等工具,让人有点崩溃。
但是在安装的过程中,Python被降版本了,从3.5降到了2.7,让人有点无奈。
试着运行一个例子,看看GPU是否被调用了
扫描二维码关注公众号,回复:
5375134 查看本文章
自动安装的cuda9.2,这个版本要求显卡驱动>=396,而我的驱动版本号是386
再次选择手动降低cuda版本,输入命令:
conda install cudatoolkit==9.0
等待安装即可。
再次运行,没有问题。
利用conda安装keras的命令:conda install keras-gpu