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

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/hung22/article/details/102750455

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

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

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

       conda create -n your_env_name python=3.7 cudnn cudatoolkit=10.0

3.进入 your_env_name 环境

       conda activate your_env_name

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

       pip install 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)