centos7下安装tensorflow-gpu版

我的版本
CUDA 8.0 +cuDAA6.0+python3.5.1+tensorflow-gpu1.4.0

说说遇到的坑(折腾了一下午):
1.查资料说是python3.6可以与其他软件兼容,于是我就安装了python3.6.6版本,结果后面测试import tensorflow时会报错ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory
网上解释:cuda未安装或者cuda的版本有问题

服务器上的CUDA 8.0 +cuDAA6.0是师兄之前装的,所以我也没有更新到9.0版本。就直接开始安装python和tensorflow-gpu

一 、python3.5.1安装

1.安装依赖
yum install openssl-devel bzip2-devel expat-devel gdbm-devel readline-devel sqlite-devel
2.下载安装包
wget https://www.python.org/ftp/python/3.5.1/Python-3.5.1.tgz

3.创建目录
mkdir /usr/local/python3
4.解压
tar -zxvf Python-3.5.0.tgz
5.编译安装

cd Python-3.5.1
./configure --prefix=/usr/local/python3
make && make install

6.创建python3和pip3的软连接

ln -s /usr/local/python3/bin/python3 /usr/bin/python3
ln -s /usr/local/python3/bin/pip3 /usr/bin/pip3

因为我之前安装过python3.6.6,创建过软连接,所以会报下面错误
ln: failed to create symbolic link ‘/usr/bin/python3’: File exists
ln: failed to create symbolic link ‘/usr/bin/pip3’: File exists
删除软连接,再重新创建就可以了
rm -rf /usr/bin/python3
rm -rf /usr/bin/pip3

7.输入python3,显示以下内容,则安装成功

$python3               
Python 3.5.1 (default, Mar  7 2019, 10:39:21) 
[GCC 4.8.5 20150623 (Red Hat 4.8.5-36)] on linux
Type "help", "copyright", "credits" or "license" for more information.

输入pip3 list可查看python3已有的包

二、安装tensorflow

1.更新pip
pip3 install --upgrade pip
不更新的话,安装tensorflow可能会报错Could not find a version that satisfies the requirement tensorflow-gpu==1.4 (from versions: ) No matching distribution found for tensorflow-gpu==1.4

2.安装tensorflow-gpu 1.4.0
pip3 install tensorflow-gpu==1.4.0

3.测试
输入python3进入交互命令
import tensorflow as tf没有报错就安装成功

测试问题:
有时候出现了错误ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory

1.检查/usr/local/cuda-8.0/lib64目录,看看libcudart.so.8.0是否存在,如果不存在,则先正确安装cuda。如果确定cuda已经存在,这个问题的出现于Linux的ldconfig命令有关。
2.输入
sudo ldconfig /usr/local/cuda-8.0/lib64
3.重新测试

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值