【20170630】TensorFlow与python学习(1)-安装

来源:互联网 发布:手机能注册淘宝邮箱吗 编辑:程序博客网 时间:2024/05/29 15:07

浙大CAPG实验室正式实习的第一天~~~


【00:背景提要】

0.1实验室工作站环境中已经配备好cudan等环境;
0.2服务器中含有python版本3.4.3,因此需安装与python兼容的TensorFlow;
0.3使用GPU加速版本。

【01:参考教程】

1.1TensorFlow中文社区
http://www.tensorfly.cn/
1.2极客学院TensorFlow社区
http://wiki.jikexueyuan.com/project/tensorflow-zh/
1.3英文官方网站:
http://tensorflow.org/
1.4官方GitHub仓库:
https://github.com/tensorflow/tensorflow
1.5中文版 GitHub 仓库:
https://github.com/jikexueyuanwiki/tensorflow-zh

【02:安装TensorFlow步骤】

2.1安装pip3:

sudo apt-get install python3-pip

2.2安装numpy:

sudo pip3 install python3 numpy

2.3安装TensorFlow:

sudo pip3 install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.2.0-cp34-cp34m-linux_x86_64.whl

2.4安装遇到问题

ImportError: libcusolver.so.8.0: cannot open shared object file: No such file or directory

解决方案:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"export CUDA_HOME=/usr/local/cuda

以上做法,只是临时设置变量 LD_LIBRARY_PATH ,下次开机,一切设置将不复存在;如何把这个值持续写到 LD_LIBRARY_PATH 里呢?

我们可以在 ~/.bashrc 或者 ~/.bash_profile 中加入 export 语句,前者在每次登陆和每次打开 shell 都读取一次,后者只在登陆时读取一次。

我的习惯是加到 ~/.bashrc 中,在该文件的未尾,可采用如下语句来使设置生效:

ls -a //查看以.开头的文件,可以看到.bashrc 文件sudo vim ./bashrc //打开文件,添加以下两行export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"export CUDA_HOME=/usr/local/cudaexit :wq //退出

修改完后,记得关掉当前终端并重新打开一个新的终端,从而使上面的配置生效。
3.测试

python 3 import tensorflow

没有报错,ok!

原创粉丝点击