使用 pip 安装 TensorFlow 时,GPU 支持所需的 CUDA 和 CuDNN 库必须单独手动安装,增加了大量负担。而使用 conda 安装 GPU 加速版本的 TensorFlow 时,只需使用命令 conda install tensorflow-gpu,这些库就会自动安装成功,且版本与 tensorflow-gpu 包兼容。此外,conda 安装这些库的位置不会与通过其他方法安装的库的其他实例产生冲突。
conda install tensorflow-gpu==1.15.0
conda install tensorflow命令只有在Linux下安装的才是集成了mkl库的TensorFlow,如果在Windows下,必须使用conda install tensorflow-mkl命令,这样才能安装集成了mkl库的TensorFlow。
参考:Intel® Optimization for TensorFlow* Installation Guide