安装清单
首先一定要看tensorflow官网
明确自己的下载配置:
我们可以看到安装tensorflow2.4.0版本需要11.0版本的CUDA,8.0.x版本的cuDNN,python版本3.6-3.8,因为你不需要源码编译就不需要管GCC了。
安装流程
1.CUDA cuda下载地址
2.cuDNN
需要注册,有账号之后可以下载 cuDNN下载地址
3.windows的CUDA安装是正常的应用安装,cuDNN安装需要设置环境变量,详细可见这篇博客
4.tensorflow-gpu
建议用anaconda首先建立环境,激活之后进入环境。
pip install tensorflow-gpu==2.4.0
安装即可
5.验证是否安装成功并且可以调用GPU
打开python后输入
import tensorflow as tf
print(tf.test.is_gpu_available())
如果是True说明你成功了!