首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

tensorflow安装GPU版本

tensorflow安装GPU版本主要要点 1.先通过该网站查看tensorflow和cuda和cudnn版本以及visual studio(MSVC)的对应关系。...(可供参考) https://www.tensorflow.org/install/source_windows#gpu 在英伟达控制面板,点击右下角的系统信息,可查看驱动的版本信息,点击上方的组件,可查看需要的...cuda的版本(这里指的是你的电脑支持的最高版本的cuda driver版本,并不是推荐你安装这个,仅参考) 选择cuda的版本比较麻烦,需要兼顾tensorflow,显卡驱动版本,电脑版本。...在安装好tensorflow-gpu后, 执行下面代码,如果打印use GPU true,则代表gpu安装完成,可以使用gpu进行训练。...__version__) print("use GPU", gpu_ok) 我的环境以及最终安装完成的版本(rtx2060驱动版本451.67,cuda10.0.130,cudnn7.6.2.24,tensorflow-gpu1.15.4

67510
您找到你想要的搜索结果了吗?
是的
没有找到

Ubuntu sever20.04 LTS版本GPU服务器深度学习环境配置

Ubuntu sever20.04GPU服务器深度学习环境配置” 在非图形界面的Ubuntu server20.04GPU服务器上配置环境,包括Nvidia驱动,cuda,cuDNN的安装,Anaconda...选择特定版本下载,复制下载链接。 6. 使用wget,下载安装包。...内核版本和cuda版本有一个对应关系,其都是基于cuda版本的系统要求,比如我的目标cuda11.4,查看其系统要求是: 一开始没太注意,我是基于ubuntu20.04 LTS安装的,并不是ubuntu20.04.2...nvidia-smi 返回以下GPU信息,说明驱动安装成功。...该界面显示的cuda版本是驱动能支持的最高版本 第一次安装后也遇到如下问题,可能是linux内核和驱动版本不匹配的原因,尝试多种方法后放弃了,按照以上流程重装成功。

2.2K20

Ubuntu18.04完美搭建Tensorflow-gpu

Ubuntu18.04完美搭建Tensorflow-GPU1.8 0.导语 这是18年的最后一个工作日,为了这篇文章耗费了半天的时间来琢磨!...然后在Ubuntu上搭建深度学习环境! 本次搭建环境,主要是在Ubuntu18.04版本,关于Centos系统搭建深度学习环境昨天也尝试了,驱动与cuda都装成功了!...第二种方式,也是我建议的方式: 首先确定自己所需要安装的驱动版本ubuntu-drivers devices 会看到: driver : nvidia-340 - distro non-free...5.Tensorflow-GPU 5.1 版本 版本tensorflow-gpu-1.8 5.2 安装 直接输入: conda install tensorflow-gpu==....最后发现,直接输入pip就可以了,所以上述换成: pip install tensorflow-gpu==. 两者区别是,cond安装会更新包,pip则不会!

3.2K40

Linux Ubuntu配置CPU、GPU通用的tensorflow

本文介绍在Linux操作系统的发行版本Ubuntu中,配置可以用CPU或GPU运行的Python新版本深度学习库tensorflow的方法。   ...在文章Anaconda配置CPU、GPU通用的tensorflow以及新版本tensorflow实现GPU加速的方法中,我们已经介绍了Windows平台下,配置CPU、GPU版本tensorflow库的方法...;而在本文中,我们就介绍一下在Linux Ubuntu环境中,CPU与GPU版本tensorflow库的配置方法。   ...本文分文两部分,第1部分为CPU版本tensorflow库的配置方法,第2部分则为GPU版本tensorflow库的配置方法;如果大家的电脑有GPU,那么就直接跳过第1部分,从本文的第2部分开始看起就好...此时,表明我们电脑中原有的NVIDIA驱动程序与新下载的版本有了冲突,导致新的版本无法正常安装。此时,我们需要在终端中,依次输入如下的代码,记得每次输入一行即可。

34530

安装tensorflow GPU版本tensorflow-gpu版本与CUDA版本对应关系(持续更新,目前到TF2.7)「建议收藏」

一、WIndows安装GPU版本tensorflow注意一下几个问题一般就不会出错 (1)确定自己要安装哪个版本tensorflow-gpu; (1)根据自己要装的tensorflow-gpu版本确定要下载的...二、tensorflow-gpu版本与CUDA版本对应关系 不同版本tensorflow-gpu与CUDA对应关系如下表所示(图片有点旧了,python版本是2.7和3.3-3.8): 对于版本号大于...1.13的tensorflow-gpu的1.x版本,如1.14、1.15,建议安装CUDA10.0,不要安装CUDA10.1,安装后会提示缺少很多库文件,而导致GPU版本tensorflow无法使用,...四、检验tensorflow-gpu安装成功 输入以下命令: import tensorflow as tf a = tf.constant([1.0,2.0,3.0],shape = [3], name...也可用一下方式临时检验: 对于2.0以前版本: tf.test.is_gpu_available() 2.1以后版本: tf.config.list_physical_devices('GPU') 发布者

5.8K20

ubuntu系统使用Anaconda安装tensorflow-gpu环境

一、环境配置版本信息: 安装tensorflow-gpu,需特别注意tensorflow-gpu、Python、CUDA、cuDNN版本的适配信息,版本不适配会导致tensorflow-gpu安装失败,...该安装教程选择的软件版本信息为:ubuntu18.04 + Anaconda3.5.3.1 + Python3.6.12 + tensorflow-gpu2.2.0 + CUDA10.1 + cuDNN7.6.5...不要直接使用默认的环境安装,最好不同任务使用不同环境,在默认环境安装新的模块可能会有冲突,导致Anaconda崩溃,最终需要卸载重装。...在默认base运行环境安装tensorflow-gpu 2.2.0的过程中,由于需更新Python的版本,会导致与原本的模块冲突而造成Anaconda崩溃。...2、tf.test.is_gpu_available()的运行结果为false: (1)首先确保tensorflow-gpu、Python、CUDA、cuDNN版本的适配信息,如果不适配,则卸载不适配的版本并重装

1.9K10
领券