Win10系统Tensorflow 2.X-GPU版本安装指南(涵盖tensorflow2.0-2.4的GPU版本)

说明:本文以2.1.0版本为例安装,亲测2.3.0版本安装为同样方法。

注:目前Tensorflow2.4版本以推出,安装方式不变,但是需要CUDA版本变更为11.0,cudNN版本变更为8.02(for CUDA11.0)

一、安装相关软件

1、 Anaconda

网址:https://www.anaconda.com/distribution/。

2、创建conda镜像

conda create -n ### python=3.X

【conda 常用命令】
升级

conda update conda
conda update anaconda
conda update anaconda-navigator    //update最新版本的anaconda-navigator   

其他

conda update -n base conda        //update最新版本的conda
conda create -n xxxx python=3.5   //创建python3.5的xxxx虚拟环境
conda activate xxxx               //开启xxxx环境
conda deactivate                  //关闭环境
conda env list                    //显示所有的虚拟环境

3、进入镜像环境安装tensorflow2.1.0

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.1.0

以及其他package,如:jupyter、pandas······

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple ###

4、安装CUDA

CUDA 10.1下载地址(必须下载10.1版本,适用于tensorflow2.3版本及以下)
即 cuda_10.1.105_418.96_win10.exe

CUDA 11.0下载地址(必须下载11.0版本,适用于tensorflow2.4版本)
即 cuda_11.0.2_451.48_win10.exe
在这里插入图片描述
在这里插入图片描述
注:Visual Studio Integration 不用安装,显卡驱动如果当前版本等于或高于CUDA10.1亦不用安装。
在这里插入图片描述

5、下载CUDNN

注意cudnn版本及其对应适用的CUDA版本号,别下错了!

cudnn7.6.5下载地址(for10.1)

cudnn8.02下载地址(for11.0)
在这里插入图片描述将解压后的4个文件在这里插入图片描述
复制到CUDA路径下
(其中cudnn765文件需要自己创建)
在这里插入图片描述

在这里插入图片描述

6、配置环境变量

右键 我的电脑属性高级系统设置
在这里插入图片描述
在这里插入图片描述
新建以下几个环境变量(若之前有CUDA其他版本的环境变量配置,则注意把新的这5个环境变量置顶,避免冲突)

在这里插入图片描述

7、安装Visual C++ Redistributable for Visual Studio for 2015、2017、2019(重要)

Visual C++ Redistributable for Visual Studio for 2019
Visual C++ Redistributable for Visual Studio for 2017
Visual C++ Redistributable for Visual Studio for 2015
如若不安装,在 import tensorflow 时会报缺少DLL组件的Error。

二、测试

conda create -n steven37 python=3.7
conda envs list

在这里插入图片描述

conda activate steven37

在这里插入图片描述

conda list

在这里插入图片描述

jupyter notebook

在这里插入图片描述

在这里插入图片描述

若果看完以上您还不会安装,那我只能

在这里插入图片描述

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值