tensorflow_gpu与CUDA、cuDNN、Python版本对应关系

版本对应列表

tensorflow的清华源wheel

tensorflow的清华源wheel列表

tensorflow_gpu安装指令

  1. 使用conda安装指定版本的tensorflow_gpu
conda install tensorflow-gpu==1.2.0
  1. 使用wheel安装指定版本的tensorflow_gpu
pip install tensorflow_gpu-2.5.0-cp38-cp38-manylinux2010_x86_64.whl

更多推荐