安装Keras - GPU 版本的方法

① conda create -n keras-gpu python=3.6
② conda install cudatoolkit=10.1
③ conda install cudnn=7.6.0
④ pip install tensorflow-gpu==2.3.0 --user 
⑤ pip install keras==2.4.3 -i https://pypi.douban.com/simple/
⑥ pip install pandas matplotlib notebook --user -i https://pypi.douban.com/simple/

你可能感兴趣的:(安装Keras - GPU 版本的方法)