在anaconda虚拟环境中安装tensorflow-gpu版本

1、首先查找你的电脑gpu配置,然后安装适合的版本。

鼠标右键,点击NVIDIA控制面板,然后点击系统信息,查看信息,然后查找对应版本。

在anaconda虚拟环境中安装tensorflow-gpu版本_第1张图片

在anaconda虚拟环境中安装tensorflow-gpu版本_第2张图片

(11条消息) 2021最新:TensorFlow各个GPU版本CUDA和cuDNN对应版本整理(最简洁)_Kakaluotuo的博客-CSDN博客_tensorflow对应的cuda版本

进入虚拟环境 ,然后首先安装对应版本的cudatoolkit和cudnn

conda  install cudatoolkit=10.1.168  (注意这里必须把版本写全,不能只写10.1)

conda install cudnn=7.6.5

pip install tensorflow-gpu==2.3.0  或者 conda install tensorflow-gpu=2.3.0

等待安装完毕

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