Centos7服务器一般用户安装tensorflow-GPU版本

7 篇文章 0 订阅
5 篇文章 1 订阅

安装tensorflow-GPU首先需要有python,cuda与cudnn,不同的tensorflow版本对应不同的cuda与cudnn,对应如下。
在这里插入图片描述

  1. 首先安装anaconda3(也可以选择python的一般版本,anaconda为python 的集成环境。(安装时,选择自动加入环境变量)
  2. 安装 cuda
  3. 下载cudnn,将cudnn中的include,lib64分别移动到cuda中的include和lib64下。
    在自己用户的 .bashrc 文件中设置环境变量,vim .bashrc
    添加

export LD_LIBRARY_PATH=/home/EXP/cuda9/lib64:$LD_LIBRARY_PATH

  1. 使用清华镜像安装tensorflow-GPU,

pip install tensorflow-gpu==1.9 -i https://pypi.tuna.tsinghua.edu.cn/simple

5.进行测试

python3
import tensorflow as tf

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值