linux下Anaconda安装tensorflow-gpu及keras其实很简单

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

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

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

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

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、重启终端,更新(This step is very importent.)

  $ 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

阅读更多
版权声明:可以转载 https://blog.csdn.net/weixin_39954229/article/details/79961172
个人分类: 深度学习
上一篇TFRecords数据格式的制作、读取及解析
博主设置当前文章不允许评论。

免费的vpn小白兔软件

2011年06月08日 978KB 下载

没有更多推荐了,返回首页

关闭
关闭