tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)

该博客详细介绍了如何在Windows10的Anaconda环境中,为Python3.8创建一个名为tensorflow-gpu的环境,并安装GPU版本的TensorFlow2.3.1,包括安装cudatoolkit10.1和cudnn7.6。通过遵循这些步骤,读者可以成功配置GPU支持的TensorFlow环境。
摘要由CSDN通过智能技术生成

本文主要讲述了在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安装完成

  • 9
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值