cuda12安装TensorFlow

注：本文假设你已经安装好了NVIDIA CUDA Toolkit和cuDNN。

1. 创建conda环境并安装TensorFlow
首先，我们需要创建一个conda环境并安装TensorFlow：

conda create -n tf-gpu tensorflow-gpu

2. 激活环境
创建完环境后，我们需要激活环境：

conda activate tf-gpu

3. 测试TensorFlow是否安装成功
安装完TensorFlow后，我们可以通过以下方式测试是否安装成功：

python
import tensorflow as tf
print(tf.__version__)

如果输出了TensorFlow的版本号，则说明安装成功。

4. 设置TensorFlow使用GPU
默认情况下，TensorFlow会使用CPU来运行计算。为了让TensorFlow使用GPU，我们需要进行如下设置：

import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

以上代码会将TensorFlow使用GPU的内存占用设置为按需分配。

5. 测试TensorFlow是否使用GPU
设置完TensorFlow使用GPU后，我们可以通过以下方式测试是否使用了GPU：

import tensorflow as tf
print(tf.test.is_gpu_available())

如果输出了True，则说明TensorFlow正在使用GPU来运行计算。

至此，我们就完成了在CUDA 12环境下安装TensorFlow的过程。