linux 安装tensorflow(gpu版本)

本文详细介绍如何在Linux环境下安装CUDA 8.0,并通过pip安装TensorFlow的不同版本(CPU版和GPU版)。同时提供了安装步骤及验证安装是否成功的测试方法。

一、安装cuda

  1.在nvidia官网下载CUDA  (linux版本cuda8.0)下载.run文件  下载好后的cuda_8.0.61_375.26_linux.run安装文件移动至当前用户根目录下 终端执行:

    

2.在安装界面中,可参照下面的描述进行选择

3.添加环境变量,安装结束后,在管理员用户根目录下,找到.bashrc文件并打开,在最后添加下面三行文本,保存退出即可

4.查看duda是否安装好

    重启终端后,执行nvcc -V,若显示以下信息,则安装cuda成功

二、安装tensorflow

 

1. 安装bcupti-dev library可执行

     sudo apt-get install libcupti-dev

2.用python直接安装

 

  首先需要pip,如果没有的话可以使用下面的命令安装
sudo apt-get install python-pip python-dev   # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n

  下面使用pip安装tensorflow

$ sudo pip install tensorflow # Python 2.7; CPU support (no GPU support)

$ sudo pip3 install tensorflow # Python 3.n; CPU support (no GPU support)

$ sudo pip install tensorflow-gpu # Python 2.7; GPU support

$ sudo pip3 install tensorflow-gpu # Python 3.n; GPU support

如: sudo pip install tensorflow==1.8.0

三、测试tensorflow是否安装成功

经过一二两个步骤,应该是安装成功了,可写个tensorflow的小程序测试一下

四 、卸载tensorflow

  $  pip uninstall tensorflow # Also add -gpu if you installed the GPU version 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值