安装TensorFlow 1.14.0

conda create -n tf114 python=3.6

pip install tensorflow-gpu==1.14.0

conda install -c anaconda cudatoolkit=10.1.243 cudnn=7.6.5

conda install cudatoolkit=10.0 

测试是否可用

import tensorflow as tf
import numpy as np
import time

# 检查TensorFlow版本
print("TensorFlow version:", tf.__version__)

# 检查是否有可用的GPU
print("GPU Available: ", tf.test.is_gpu_available())