安装tensorflow 的过程简介
发布日期:2021-06-30 17:15:51 浏览次数:2 分类:技术文章

本文共 2494 字,大约阅读时间需要 8 分钟。

 安装tensorflow 的过程简介:

1、系统环境

ubuntu 16.04 x64 desktop 

2、python环境

pip环境为9.0.1

如果pip不是9.0.1 请运行下面的命令:

pip install --upgrade pip

3、cuda的环境为7.5,请不要安装8.0的。

因为tensorflow-0.12.0rc0-cp27-none-linux_x86_64.whl,需要的库为libcudart.so.7.5,而不是libcudart.so.8.0的。

4、具体的不用去那么复杂,只需要自己下载对应的deb文件就可以安装了。

中间走了很多弯路,再次将deb的安装包的下载地址发布一下:

这个就是cuda7.5 下载所需要的地址:

主要下载以下几个文件:

1)、cuda-license-7-5_7.5-18_amd64.deb

2)、cuda-core-7-5_7.5-18_amd64.deb

3)、cuda-cudart-7-5_7.5-18_amd64.deb

4)、cuda-cudart-dev-7-5_7.5-18_amd64.deb

5)、cuda-driver-dev-7-5_7.5-18_amd64.deb

上面的下载地址上都有这个几个文件,下载后,按照顺序安装:

dpkg --install cuda-license-7-5_7.5-18_amd64.deb dpkg --install cuda-driver-dev-7-5_7.5-18_amd64.deb dpkg --install cuda-cudart-dev-7-5_7.5-18_amd64.deb dpkg --install cuda-cudart-7-5_7.5-18_amd64.deb dpkg --install cuda-cudart-dev-7-5_7.5-18_amd64.deb

安装完成后,就可以在/usr/local/cuda-7.5目录下看到对应的安装库了。

5、安装tensorflow,这个直接用pip安装,但最好修改下pip源,不然速度很慢的。

直接运行如下命令:

sudo pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl  --trusted-host pypi.douban.com

如果需要安装其他的,比如six,也可以直接用douban的pip源

pip install six --upgrade --target="/usr/lib/python2.7/dist-packages" --trusted-host pypi.douban.com

6、找不到libcudart.so.7.5库的问题:

libcudart.so.7.5: cannot open shared object file: No such file or directory。

这个可以直接copy到对应的目录,也可以修改/etc/profile

解决方法mark以下几种方法:
1)ldconfig
32-bit:sudo ldconfig /usr/local/cuda-7.5/lib
64-bit:sudo ldconfig /usr/local/cuda-7.5/lib64
2)设置.brshrc文件或/etc/profile文件
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-7.5/lib64"
soucre ~/.bashrc 或 source /etc/profile
3)临时生效的设置
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-7.5/lib64"

最直接的还是第一个方法,直接copy到lib目录下;

cp /usr/local/cuda-7.5/lib64/libcudart.so.7.5 /usr/local/lib/libcudart.so.7.5 && sudo ldconfig

7、运行过程:

root@eewiki:~# pythonPython 2.7.12 (default, Nov 19 2016, 06:48:10)[GCC 5.4.0 20160609] on linux2Type "help", "copyright", "credits" or "license" for more information.>>> import tensorflowI tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcublas.so locallyI tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcudnn.so locallyI tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcufft.so locallyI tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcuda.so.1 locallyI tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcurand.so locally>>> exit()

转载地址:https://leekwen.blog.csdn.net/article/details/77091748 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:Burp Suite中proxy的简单用法
下一篇:安装tensorflow 的过程简介

发表评论

最新留言

关注你微信了!
[***.104.42.241]2024年04月07日 19时19分25秒