linux下使用Anaconda安装tensorflow-gpu及keras

之前网上看了很多教程,走了很多弯路,感觉有必要记一下,cuda、cudnn也都不需要自己装,conda会在装tensorflow-gpu时帮你搞定,省了很多事,具体步骤如下:

一、安装驱动(最简单的安装驱动方法)

 

1、点桌面左上角搜索本机程序的图标,找到“附加驱动”

linux下使用Anaconda安装tensorflow-gpu及keras_第1张图片

2、在“附加驱动”里,系统会自动搜索N卡驱动,列表里会提供对应你显卡的最新版官方驱动。

linux下使用Anaconda安装tensorflow-gpu及keras_第2张图片

3、最后点“应用更改”,等待安装完毕。

4、重启系统即可。

参考:https://jingyan.baidu.com/article/60ccbceb50803c64cab1978c.html

二、安装Anconda,并更新

1、https://www.anaconda.com/download/#linux 下载

2、cd到文件下载的位置输入如下命令安装Anaconda

 $ bash Anaconda3-5.1.0-Linux-x86_64.sh

3、重启终端,更新

  $ conda update -n base conda

三、安装tensorflow-gpu及keras

 

1、创建一个环境,指定python版本

$ conda create -n tf python=3.6

2、激活创建的tf环境:

$ conda activate tf 或者 $ source activate tf

3、在该环境下安装tensorflow-gpu及keras框架:

$ conda install tensorflow-gpu keras

4、查看安装是否成功

$ python>>import keras  import tensorflow

5、安装其他所需要的库

 

$ conda install pillow$ conda install matplotlib$ conda install scikit-learn$ conda install graphviz pydot

6、退出该环境

$ conda deactivate  或者  source deactivate

7、移除tf环境

$ conda remove -n tf --all

8、查看GPU运行状况

$ watch -n 0 nvidia-smi

你可能感兴趣的:(深度学习)