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)

ubuntu下用conda快速安装tensorflow-gpu_第1张图片

突然发现这个命令可以自动安装cuda和cudnn等工具,让人有点崩溃。

ubuntu下用conda快速安装tensorflow-gpu_第2张图片

但是在安装的过程中,Python被降版本了,从3.5降到了2.7,让人有点无奈。

ubuntu下用conda快速安装tensorflow-gpu_第3张图片

试着运行一个例子,看看GPU是否被调用了

ubuntu下用conda快速安装tensorflow-gpu_第4张图片

自动安装的cuda9.2,这个版本要求显卡驱动>=396,而我的驱动版本号是386

再次选择手动降低cuda版本,输入命令:

conda install cudatoolkit==9.0

ubuntu下用conda快速安装tensorflow-gpu_第5张图片

等待安装即可。

ubuntu下用conda快速安装tensorflow-gpu_第6张图片

再次运行,没有问题。

利用conda安装keras的命令:conda install keras-gpu

 

 

你可能感兴趣的:(ubuntu下用conda快速安装tensorflow-gpu)