ubuntu20搭配RTX3090安装tensorflow2.4

huoxingwen

已于 2022-02-26 21:01:35 修改

阅读量756

点赞数

分类专栏： tensorflow 文章标签： tensorflow anaconda gpu python 深度学习

于 2021-04-28 12:23:03 首次发布

本文链接：https://blog.csdn.net/qq_36711448/article/details/116230834

版权

tensorflow 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

本文详细介绍了如何在Anaconda环境中安装TensorFlow 2.4，包括使用pip而非conda install，以及正确配置CUDA和CUDNN。通过实例演示了GPU与CPU性能测试，旨在帮助读者快速避开安装过程中的常见问题。

摘要由CSDN通过智能技术生成

遇坑太多了，方法反而最简单的最快

1.安装专有的460驱动，run文件安装坑太多了

2.在conda里面安装tensorlfow2.4

先用pip装tf，不要conda install tensorflow-gpu，是个坑

conda create -n tf24 python=3.8
conda activate tf24
pip install tensorflow

3.安装cuda和cudnn，版本一定要对应

去anaconda 搜索对应cuda cudnn版本

安装CUDA11

conda install -c anaconda cudatoolkit

安装cudnn8

conda install -c nvidia cudnn

4.测试

import tensorflow as tf
import timeit
 
n=100000000
with tf.device('/cpu:0'):                 
    cpu_a = tf.random.normal([10,n])     
    cpu_b = tf.random.normal([n,10])
print(cpu_a.device,cpu_b.device)
with tf.device('/gpu:0'):
    gpu_a = tf.random.normal([10,n])
    gpu_b = tf.random.normal([n,10])
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=100)
gpu_time = timeit.timeit(gpu_run,number=100)
print('compute:',cpu_time,gpu_time)

结果