深度学习环境搭配——cuda+cudnn+tensorflow-gpu+keras

Neysa ·
更新时间:2024-09-20
· 777 次阅读

深度学习环境搭配——cuda+cudnn+tensorflow-gpu+keras 一、手动安装:

cuda下载:https://developer.nvidia.com/cuda-toolkit
cudnn下载:https://developer.nvidia.com/cudnn
tensorflow版本和cuda的对应关系:

https://www.tensorflow.org/install/source#common_installation_problems
https://www.tensorflow.org/install/source_windows
部分版本对应关系
keras和tensorflow的对应关系:https://docs.floydhub.com/guides/environments/ 二、自动安装:

1、下载Anaconda:
2、创建一个环境
3、添加清华镜像(参考https://blog.csdn.net/qq_27825451/article/details/89237574)

#添加清华的源(https://mirror.tuna.tsinghua.edu.cn/help/anaconda/)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

4、conda install tensorflow-gpu==x.xx


作者:旧月照新人



环境 学习 cuda cudnn 深度学习 tensorflow keras

需要 登录 后方可回复, 如果你还没有账号请 注册新账号
相关文章