conda tf2 gpu 安装

最新推荐文章于 2023-09-28 19:35:27 发布

阿豪boy

最新推荐文章于 2023-09-28 19:35:27 发布

阅读量151

点赞数

文章标签：深度学习 tensorflow 神经网络快速排序机器学习

本文链接：https://blog.csdn.net/qq_35516360/article/details/122065442

版权

本文档详细介绍了如何在conda环境中安装TensorFlow2的GPU版本，首先需要激活相应的conda环境，然后进行安装步骤。安装完成后，通过测试确保TensorFlow2 GPU版已成功配置并可以使用。

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

原文链接: conda tf2 gpu 安装

上一篇: js 快速排序递归实现

下一篇: js 原型链污染和预防

安装

conda create -n TF_2G python=3.6 anaconda

激活环境

conda activate TF_2G

安装

conda install cudatoolkit=10.0 cudnn
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

测试


import tensorflow as tf
 
X = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
y = tf.constant([[10.0], [20.0]])
 
 
class Linear(tf.keras.Model):
    def __init__(self):
        super().__init__()
        self.dense = tf.keras.layers.Dense(
            units=1,
            activa