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

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

1.打开Anaconda Prompt
tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第1张图片
2.输入下面命令,利用Anaconda创建一个python3.8的环境,环境名称为tensorflow-gpu(名称任意)

conda create -n tensorflow-gpu python=3.8

tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第2张图片
输入y,并按回车键
tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第3张图片

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

conda activate tensorflow-gpu

tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第4张图片
4.输入下面命令,安装python的pip工具

python -m pip install --upgrade pip

tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第5张图片
5.输入下面命令,安装TensorFlow2.3.1
(切换到阿里云进行安装)

pip install tensorflow-gpu==2.3.1 -i https://mirrors.aliyun.com/pypi/simple/

tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第6张图片

等待安装完成

6.输入下面命令,安装英伟达的cudatoolkit 10.1

conda install cudatoolkit=10.1

tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第7张图片
输入y,等待安装完成
tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第8张图片
cudatoolkit 10.1安装完成
tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第9张图片
7.输入下面的命令行,安装英伟达cudnn 7.6

conda install cudnn==7.6.*

tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第10张图片
输入y,并等待安装完成
tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第11张图片
8.进入python环境,输入:

python

输入命令导入tensorflow库

import tensorflow as tf

输入命令查看python版本

tf.__version__

tensorflow安装步骤(GPU版本,Anaconda环境下,Windows10)_第12张图片
tensorflow2.3.1安装完成

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