Tensorflow的GPU版本安装＜24.3.22更新＞

Tensorflow2.10 and Python3.9

今天重装了一下环境，记录一下

nvidia-smi命令显示的CUDA版本（不须卸载重装）

1. 创建独立环境并激活

conda create -n py39 python=3.9
conda activate py39

2. 安装相关软件包

conda install tensorflow=2.10
pip install tensorflow-gpu==2.10 --user      # 注:我不加`--user`会报错
conda install keras=2.10
conda install cudnn

如果conda命令用不了，就用pip

3. 版本对应

如果不安装tensorflow-gpu 以下命令只显示CPU 不显示GPU

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

4. 测试

不用gpu，cifar-10每epoch需要13s左右，使用后如下