ubuntu16.04+cuda8.0+cudnn5.1 安装

最新推荐文章于 2025-02-21 17:29:10 发布

原创最新推荐文章于 2025-02-21 17:29:10 发布 · 966 阅读

CC 4.0 BY-SA版权

文章标签：

3 篇文章

订阅专栏

本文介绍如何在Ubuntu 16.04系统上安装带有GPU支持的TensorFlow 1.1.0版本，包括解决权限问题的方法，并提供安装命令。此外，还分享了使用GPU加速神经风格迁移项目的体验。

部署运行你感兴趣的模型镜像

Ubuntu16.04 cuda8.0 cudnn5.1

pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.1.0-cp27-none-linux_x86_64.whl（中文社区的版本都特别低，此处需要注意）

如果遇到权限问题，请使用完整路径

sudo -H xxxx/anaconda2/bin/pip install xxxxxx.whl
https://pypi.python.org/pypi/tensorflow 下载最新即可

这里写图片描述

OK～大功告成！
这里写图片描述
显示调用GPU：0.

PS:官网中提示的配置 TensorFlow 的 Cuba 选项
从源码树的根路径执行:
xxxxxx
这个地方已经配置过了应该，就是两个库和默认路径。所以可以不管这里。

感动得要哭，确实比CPU要快很多！～

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。它提供了一个灵活的平台,用于构建和训练各种机器学习模型