前言:首先需要具备anaconda
1、创建虚拟环境:conda create -n 你的名字 python=3.6
(python版本可自己指定,但3.6似乎兼容性最好)
2、安装cuda与cudnn:
cuda部分: conda install cudatoolkit=10.1
cudnn部分: conda install cudnn=7.6.5
3、使用国内镜像源安装tensorflow-gpu:
pip install -i https://mirrors.aliyun.com/pypi/simple tensorflow-gpu==2.1
后记:tensorflow 与 cuda、cudnn 版本之间有着对应关系,如下图所示,版本不一致会不兼容,详情可以去tensorflow官网查看