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

TensorFlow GPU 版安装

0x00 前言 CPU版的TensorFlow安装还是十分简单的,也就是几条命令的时,但是GPU版的安装起来就会有不少的坑。在这里总结一下整个安装步骤,以及在安装过程中遇到的问题和解决方法。...整体梳理 安装GPU版的TensorFlow和CPU版稍微有一些区别,这里先做一个简单的梳理,后面有详细的安装过程。...Python NVIDIA Cuda cuDNN TensorFlow 测试 0x01 安装Python 这里有两种安装的方法: 安装基本的Python环境,需要什么再继续安装。...下载下来的文件就deb包,直接dpkg -i 安装即可。 0x04 安装TensorFlow Tensorflow官网中有很详细的说明。.../storage.googleapis.com/tensorflow/linux/gpu/tensorflow-1.0.1-cp27-cp27m-linux_x86_64.whl 0x05 验证安装 这里跑一个小例子来验证一下

1.4K30

tensorflow安装GPU版本

tensorflow安装GPU版本主要要点 1.先通过该网站查看tensorflow和cuda和cudnn版本以及visual studio(MSVC)的对应关系。...(可供参考) https://www.tensorflow.org/install/source_windows#gpu 在英伟达控制面板,点击右下角的系统信息,可查看驱动的版本信息,点击上方的组件,可查看需要的...在安装后cuda后可以通过cmd输入nvcc -V验证一下。 在安装好tensorflow-gpu后, 执行下面代码,如果打印use GPU true,则代表gpu安装完成,可以使用gpu进行训练。...import tensorflow as tf gpu_ok = tf.test.is_gpu_available() print("tf version:", tf....__version__) print("use GPU", gpu_ok) 我的环境以及最终安装完成的版本(rtx2060驱动版本451.67,cuda10.0.130,cudnn7.6.2.24,tensorflow-gpu1.15.4

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

    不安装tensorflow-gpu如何使用GPU

    这是个很严峻的问题,每次跑代码,内存就炸了,gpu还没开始用呢,看一些博客上是这样说的: 方法一: import os os.environ["CUDA_VISIBLE_DEVICES"] = "2"#...方法二: 卸载cpu版本的tensorflow,重新安装gpu版本的 好不容易装上的,如果可以用其他的方法,那么我还是想试一下的。...方法三: 正在探讨中,找到了再补充在这个博客中 还有一个很有意思的是,你怎么知道你的某个环境用的是cpu还是gpu: 我引用一下,原文出自https://blog.csdn.net/weixin_37251044.../article/details/79790270 import numpy import tensorflow as tf a = tf.constant([1.0,.../job:localhost/replica:0/task:0/device:GPU:0 MatMul: /job:localhost/replica:0/task:0/device:GPU

    1.8K30

    Tensorflow-gpu的安装+pycharm

    文章目录 一、基本步骤 1.安装Anaconda 2.在Anaconda中创建环境 3.安装tensorflow 4.在pycharm中配置环境 二、开始安装 1.安装Anaconda 2.在Anaconda...中创建环境 安装tensorflow 4.在pycharm中配置环境 一、基本步骤 1.安装Anaconda 2.在Anaconda中创建环境 3.安装tensorflow 4.在pycharm中配置环境...在最前面 这就表明了进入了环境 安装tensorflow 接着在新建好的环境里面输入 pip install tensorflow-gpu 或 conda install tensorflow-gpu...=2.0.0 如果速度慢的话也可以用镜像源安装 python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu...__version__ gpu_ok = tf.test.is_gpu_available() print("tf version:",version,"\nuse GPU",gpu_ok) 验证是否安装成功

    32810

    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...2.2.0: pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple 4、安装CUDA 10.1:...true,否则为false tf.test.is_built_with_cuda() # 检查tensorflow是否可以获取到GPU,安装成功则显示true,否则为false tf.test.is_gpu_available...在默认base运行环境安装tensorflow-gpu 2.2.0的过程中,由于需更新Python的版本,会导致与原本的模块冲突而造成Anaconda崩溃。

    2.1K10
    领券