栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 系统运维 > 运维 > Linux

tensorflow GPU安装

Linux 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

tensorflow GPU安装

想安装,直接看后面的 TF 安装

    需要装的软件CUDA驱动,cuDNN,TF
CUDA驱动,cuDNN,TF 关系介绍

https://blog.csdn.net/xs11222211/article/details/84782046

除TF外都按当期那最新版本

    Ubuntu版本决定最高显卡驱动版本
    查看当前系统推荐的显卡驱动版本

    ubuntu-drivers devices 
    

    驱动版本决定最高cuda和cudnn版本
    显卡版本对应的cuda和cudnn版本
    https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

    cuda版本决定TF最高版本
    https://www.tensorflow.org/install/source#install_gpu_support_optional_linux_only

TF安装

建议 用 docker 安装在 tensorflow 的 docker hub找直接版本的镜像运行也可以使用镜像提供的cuda等环境,直接在conda里用虚拟环境搭TF

conda create --name tensorflow_gpu python=3.6
source activate tensorflow_gpu
conda install tensorflow-gpu==1.11.0
检验TF 安装情况
python -c "import tensorflow as tf; print(tf.test.gpu_device_name())"

or

import tensorflow as tf
tf.test.is_gpu_available()
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/734446.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号