在Windows 10 使用 Anaconda安装 TensorFlow gpu 2.0 的步骤(不需在nivida网站,下载cuda driver)

首先使用Driver_Booster去更新, 你的NVIDIA显示适配器驱动.

在 Windows 10 使用 Anaconda安装 TensorFlow gpu 2.0 的步骤:

1.下载并安装 Anaconda
https://www.anaconda.com/distribution/

2.安装Anaconda 后, 打开Anaconda Prompt输入.

  conda create -n your_env_name python=3.7

例如:   

conda create -n py3-tfg2 python=3.7 

3.进入 your_env_name 环境

       conda activate your_env_name

例如:

conda activate py3-tfg2

4.然后输入, 安装Tensorflow-gpu 2.0。

      

conda install -c anaconda tensorflow-gpu 

那就安装 TensorFlow-gpu 2.0完毕了。

 

可以用以下的test.py 测试一下 ( 这个例子是从网上引用)

import tensorflow as tf

import timeit

with tf.device('/cpu:0'):

    cpu_a =tf.random.normal([10000,1000])

    cpu_b =tf.random.normal([1000,2000])

    print(cpu_a.device,cpu_b.device)


with tf.device('/gpu:0'):

    gpu_a =tf.random.normal([10000,1000])

    gpu_b =tf.random.normal([1000,2000])

    print(gpu_a.device,gpu_b.device)


def cpu_run():

    with tf.device('/cpu:0'):

        c = tf.matmul(cpu_a,cpu_b)

        return c

def gpu_run():

    with tf.device('/gpu:0'):

        c = tf.matmul(gpu_a,gpu_b)

        return c


cpu_time = timeit.timeit(cpu_run, number=10)

gpu_time = timeit.timeit(gpu_run, number=10)

print('warmup: ',cpu_time,gpu_time)


cpu_time = timeit.timeit(cpu_run, number=10)

gpu_time = timeit.timeit(gpu_run, number=10)

print('run time: ',cpu_time,gpu_time)