前提:已有anaconda
1、创建虚拟环境
conda create -n tf python=3.7
2、安装cuda和cudnn
进入虚拟环境
conda activate tf
安装cudatoolkit=10.0 cudnn=7.6.5 (稍微大于7.4也可)
conda install -c conda-forge cudatoolkit=10.0 cudnn=7.6.5
3、安装tensorflow1.15
pip install tensorflow-gpu==1.15
Note:
一个tensorflow版本与cuda、cudnn的兼容表
https://www.tensorflow.org/install/source#gpu