linux安装TensorFlow-GPU版本 非常详细安装必看

本文详述了在Linux系统中安装TensorFlow-GPU 1.14.0的步骤,包括确认系统环境、安装NVIDIA驱动、CUDA 10.0和CUDNN,以及设置环境变量。安装过程中需注意CUDA和CUDNN版本与TensorFlow的对应关系,最后通过Python验证TensorFlow-GPU是否成功安装。
摘要由CSDN通过智能技术生成

Linux安装TensorFlow-GPU==1.14.0版本

1.首先确保linux环境干净以及没有tensorflow
2.常规安装TensorFlow方法 直接使用pip3 install tensorflow -i https://pypi.douban.com/simple安装即可

一:
首先 登入NVIDIA查找相应的GPU驱动
查找相应gpu驱动
此处选择自己的显卡参数型号  我的是特斯拉 p系列 cuda是10.0的
然后点击search
点击search进入查询结果界面 在这里可以看到相对应的显卡驱动版本   如图所示
这里就是相应的显卡驱动Version
然后点击DownLoad进入下载界面
点击我同意(Agree&DownLoad)接下来就可以自动下载了
下载的驱动名称示例 NVIDIA-Linux-x86_64-410.104.run

安装的话就很简单了
sh NVIDIA-Linux-x86_64-410.104.run

输入命令验证是否安装成功
[root@ctnr ~]# nvidia-smi
Tue Mar 26 11:31:34 2019
±----------------------------------------------------------------------------+
| NVIDIA-SMI 410.104 Driver Version: 410.104 CUDA Version: 10.0 |
|-------------------------------±---------------------±---------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|=++==============|
| 0 Tesla P4 Off | 00000000:00:08.0 Off | 0 |
| N/A 28C P8 7W / 75W | 0MiB / 7611MiB | 0% Default |
±------------------------------±---------------------±---------------------+

±----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|========================&#

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值