Chipmunk & Panda

-- 鼠熊部落格

All work and no play makes Jack a dull boy.

Windows 下使用 Anaconda 安装 TensorFlow 2.3 并支持 GPU

  备忘    

方法来自：Why is Tensorflow not recognizing my GPU after conda install?

conda install tensorflow-gpu 语句目前有 BUG，不会安装 GPU 所需 CUDA 套件，即使之后手动安装 CUDA 套件也无效。

方法 1：从低版本升级

先安装低版本 TensorFlow，再将其升级。

1
2
conda install tensorflow-gpu=2.1
pip install tensorflow-gpu==2.3.1

方法 2：指定具体安装包

BUG 原因可能是 Anaconda 在安装时选择了有错误的 TensorFlow 包，所以通过指定具体的正确安装包来解决问题。

  • Python 3.7：conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py37h936c3e2_0
  • Python 3.8：conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py38h1fcfbd6_0