找到自己对应的不同的版本的tensorflow-gpu,cuda,cudnn
cuda链接:https://developer.nvidia.com/cuda-toolkit-archive
cudnn链接:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse760-101
1、创建虚拟环境:conda create -n 你的虚拟环境名字 python=你想要的python版本
例如:
conda create -n huanjing python=3.6
2、激活虚拟环境: activate 虚拟环境名字
windows下:
activate huanjing
ubuntu下:
source activate huanjing
3、安装自己的tensorflow-gpu版本:
例如:
pip install tensorflow-gpu=2.2.0
4、安装你tensorflow-gpu对应的cuda和cuDNN: conda install cudatoolkit=你tensorflow要安装的版本所对应的cuda版本
conda install cudatoolkit=10.1
conda install cudnn=7.6
注意:此处要加上版本号,否则安装会出现版本问题