Win11 下通过Anaconda安装tensorflow

1.创建虚拟环境并激活

创建虚拟环境:

注意python别安装太高版本的,有可能tensorflow不支持

conda create -n tensorflow_learning python=3.6

选择虚拟环境激活:

activate tensorflow_learning

2.安装tensorflow (GPU版本)

首先查看显卡型号

然后去英伟达官网查看显卡计算能力

环境搭建04——如何选择安装的TensorFlow版本?_命名无能的博客-CSDN博客_tensorflow版本

安装CUDA遇到问题,参考下文解决:

CUDA安装失败-Nsight compute安装失败-如何测试CUDA是否安装成功?Reason: VS** was not found_dssgresadfsrgre的博客-CSDN博客_cuda 组件未安装

然后安装cuDNN,解压出来以后把文件夹里的东西拷贝到CUDA对应文件夹里。cuDNN下载

 安装tensorflow:

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

经测试,安装成功!

 中间踩了好多坑……

最后跑以前的大作业时,发现keras有点问题,google后将 Keras 降级到 2.6.0,即与 TensorFlow 相同的版本,问题解决。

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

猜你喜欢

转载自blog.csdn.net/weixin_44709392/article/details/123779329