conda 已经为我们安装软件提供了非常大的便利。
如果用conda安装TensorFlow,他会自动给我们安装匹配好的cuda以及cudnn。注意cuda,在conda中是以cudatoolkit名称存在的。
conda 安装tf
但是conda的库一般不全,一般新版的tf是没有的,所以有的时候我们需要用pip来安装新版的tf包。例如
conda install cudatoolkit=11.0
conda install cudnn=8.2.1
pip install tensorflow==2.6.0
这三条命令是没有先后顺序的,可以先安装tf,然后测试tf是否能够正常调用gpu,不能调用就会通过报错来告诉我们它需要什么版本的cuda以及cudnn。
例如下图中测试tf时候报错