本文主要讲述了在Anaconda环境下,GPU版本tensorflow2.3.1的安装步骤,Windows10系统中Anaconda的安装步骤可以阅读此篇博客:
Anaconda安装步骤(Windows10)
1.打开Anaconda Prompt

2.输入下面命令,利用Anaconda创建一个python3.8的环境,环境名称为tensorflow-gpu(名称任意)
conda create -n tensorflow-gpu python=3.8

输入y,并按回车键

3.输入下面命令,进入名称为tensorflow-gpu的环境中去(名称任意)
conda activate tensorflow-gpu

4.输入下面命令,安装python的pip工具
python -m pip install --upgrade pip

5.输入下面命令,安装TensorFlow2.3.1
(切换到阿里云进行安装)
pip install tensorflow-gpu==2.3.1 -i https://mirrors.aliyun.com/pypi/simple/

等待安装完成
6.输入下面命令,安装英伟达的cudatoolkit 10.1
conda install cudatoolkit=10.1

输入y,等待安装完成

cudatoolkit 10.1安装完成

7.输入下面的命令行,安装英伟达cudnn 7.6
conda install cudnn==7.6.*

输入y,并等待安装完成

8.进入python环境,输入:
python
输入命令导入tensorflow库
import tensorflow as tf
输入命令查看python版本
tf.__version__

tensorflow2.3.1安装完成
该博客详细介绍了如何在Windows10的Anaconda环境中,为Python3.8创建一个名为tensorflow-gpu的环境,并安装GPU版本的TensorFlow2.3.1,包括安装cudatoolkit10.1和cudnn7.6。通过遵循这些步骤,读者可以成功配置GPU支持的TensorFlow环境。
&spm=1001.2101.3001.5002&articleId=122907282&d=1&t=3&u=a20c9fea7389455fab871966aeda180d)
3243

被折叠的 条评论
为什么被折叠?



