安装tensorflow_gpu版本

tensorflow gpu安装流程如下

conda create -n tfgpu3.9 python=3.9.21

conda activate tfgpu3.9

conda install -c conda-forge numpy=1.23.5 scipy=1.9.3 scikit-learn=1.2.2

conda install -c conda-forge cudatoolkit=11.2.2 cudnn=8.1.0

pip install tensorflow==2.10.1

conda install ipykernel

参考的链接如下

原文链接