最简单的安装tensorflow-gpu的方法
利用conda安装(如果没有可以先安装Anaconda)
##conda create -n 你的虚拟环境名字 python=你想要的python版本##
conda create -n tensorflow python=3.7
##source activate 虚拟环境名字 ##这一步是进入你的虚拟环境##
source activate tensorflow
##以下尽量在你创建的虚拟环境中进行##
##conda install cudatoolkit=你tensorflow要安装的版本所对应的cuda版本
conda install cudatoolkit=10.0
conda install cudnn
##conda install tensorflow-gpu=你想要的版本
conda install tensorflow-gpu=2.0
一定要注意cuda和cudnn的版本对应好你的tensorflow
我感觉这个方法很好用,不用动整个系统环境变量,只需要在虚拟环境中安装,所以可以不同的虚拟环境配置不同的tensorflow,pytorch等。而且不需要管理员权限。
如果你同时需要tensorflow1.14和1.13,就可以开两个虚拟环境,分别安装不同版本的cuda和cudnn,再安装对应的tensorflow13或14就好了