Ubuntu 18下安装tensorflow(基于python3.6)

本文详细介绍了在Ubuntu 18.04上安装TensorFlow-GPU 1.5所需的步骤,包括安装NVIDIA驱动、CUDA 9.0、cuDNN 7.1,并确保所有组件兼容。通过检查和测试,确认TensorFlow已成功利用GPU运行。
摘要由CSDN通过智能技术生成

#配置如下
Ubuntu 18.04 LTS

python 3.6

nvidia driver 390

cuda 9.0

cudnn 7.1

tensorflow 1.5

安装显卡驱动

查看NVIDIA driver 推荐。

$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C81sv00001028sd000011C0bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP107 [GeForce GTX 1050]
driver   : nvidia-driver-396 - third-party free recommended
driver   : nvidia-driver-390 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

安装:

$ sudo ubuntu-drivers autoinstall

按照推荐的驱动安装,如果失败,请添加NVIDIA driver源仓库然后在安装,安装命令如下:

$ sudo add-apt-repository ppa:graphics-drivers/ppa

测试安装是否成功(提示无法找到的时候,可以先试试重启机器再测试)

$ nvidia-smi

Tue May 22 12:14:56 2018
±----------------------------------------------------------------------------+
| NVIDIA-SMI 390.59 Driver Version: 390.59 |
|-------------------------------±---------------------±---------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|=++==============|
| 0 GeForce GTX 1050 Off | 00000000:01:00.0 Off | N/A |
| 20% 39C P0 N/A / 75W | 444MiB / 200

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值