Win10系统Tensorflow 2.X-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版本号,别下错了!
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
若果看完以上您还不会安装,那我只能