本文共 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 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!