ubuntu下用conda快速安装tensorflow-gpu

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

猜你喜欢

转载自blog.csdn.net/weixin_42718092/article/details/86022243