Anaconda环境下 tensorflow keras gpu 版本环境安装

conda install tensorflow-gpu keras

会导致同时安装tensorflow 的cpu和gpu版本

 

可以使用pip安装

1.安装tensorflow-gpu 和keras

pip install tensorflow-gpu
pip install keras

tensorflow-gpu >1.13.0

keras >2.2.0

2.安装cudatoolkit 和 cudnn

直接使用以下语句安装,可能会出现cuda的版本过低

conda install cudatoolkit
conda install cudnn

可以使用语句,安装cudatoolkit 10.0版本 和cudnn7.6版本,可在https://anaconda.org搜索具体的conda可安装的库以及相应的安装的语句

conda install -c anaconda cudatoolkit 
conda install -c anaconda cudnn 

 

你可能感兴趣的:(Anaconda环境下 tensorflow keras gpu 版本环境安装)