安装GPU的tensorflow(2024年安装)

1.看python版本

按win+R，输入cmd，进入控制台，输入python，看python版本号

or 在Anaconda Prompts输入python，看python版本号

2.看CUDA版本

打开NVIDlA Control Panel ->系统信息->组件

 3.在Anaconda Prompt输入

conda create -n py38 python=3.8(3.10目前tensorflow没有适用的，可在官网查看，我在pycharm看到我有python3.8的)

打开tensorflow官网

在 Windows 环境中从源代码构建  |  TensorFlow (google.cn)

-n:自定义的虚拟环境名，我的虚拟环境为py38；
后选定python版本，选择python 3.8；

激活环境

conda search cuda
查看cuda版本号

conda search cuda

查看cudnn版本号

conda search cudnn

输入conda install cudatoolkit=10.1.243 等待安装，完毕后装cudnn

输入conda install cudnn=7.6.5 等待安装，完毕后装tensorflow-gpu

输入pip install tensorflow-gpu==2.3.0 测试是否安装成功，等待有点久，安装成功

测试一下

输入python，导入包import tensorflow as tf，报错

pip install protobuf==3.19.6

 重新测试，进入python后，输入以下命令

import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())

安装完毕