Windows11+CUDA11.1安装Tensorflow-gpu

从tensorflow的官网可以看到是没有CUDA11.1对应的CUDA版本的:
Windows11+CUDA11.1安装Tensorflow-gpu_第1张图片
因此,直接安装tensorflow==2.6.0试一下(这里我的python=3.8):

pip install tensorflow-gpu==2.6.0 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

推荐使用豆瓣源。

测试tensorflow是否安装成功时报错:

python
import tensorflow as tf

Windows11+CUDA11.1安装Tensorflow-gpu_第2张图片
解决:安装低版本的protoc:

pip uninstall protoc
pip install protoc==3.19.6 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

然后测试成功安装:

python
import tensorflow as tf
tf.__version__
tf.test.is_gpu_available()
tf.config.list_physical_devices('GPU')

输出如下:
Windows11+CUDA11.1安装Tensorflow-gpu_第3张图片

你可能感兴趣的:(tensorflow,python,人工智能)