tranflow-gpu的安装

首先我用的是
本机的环境：
win11
DIrver Version 527.99
CUDA版本 12.0
image

python3.8
cuda_11.6.0_511.23
cudnn-windows-x86_64-8.4.0.27_cuda11.6-archive
image
安装完之后我们创建一个python3.8的虚拟环境，然后我们进入这个环境
输入

pip install tensorflow-gpu==2.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

之后就安装好了，但是我执行的时候出了一个bug
image
解决办法
点击ternimal输入pip uninstall protobuf将现有版本的卸载，然后输入pip install protobuf==3.19.0即可。
然后测试一下

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

image

posted @ 2023-06-14 15:47  lipu123  阅读(70)  评论(0)  编辑  收藏  举报