Windows11+CUDA11.1安装Tensorflow-gpu

Goafan

已于 2023-05-21 15:31:12 修改

阅读量1.5k

点赞数 2

文章标签： tensorflow python 人工智能

于 2023-05-21 15:30:37 首次发布

本文链接：https://blog.csdn.net/weixin_45231460/article/details/130792382

版权

在Tensorflow官网找不到CUDA11.1对应的版本，尝试安装tensorflow==2.6.0并使用python3.8。遇到导入错误后，卸载并安装低版本的protoc=3.19.6解决了问题。测试显示Tensorflow成功安装并能识别GPU。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

从tensorflow的官网可以看到是没有CUDA11.1对应的CUDA版本的：
在这里插入图片描述
因此，直接安装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

在这里插入图片描述
解决：安装低版本的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')

输出如下：
在这里插入图片描述