admin管理员组文章数量:1323775
喜大普奔!!! 使用anaconda直接安装深度学习环境所需的cuda、cudnn、tensorflow-gpu,windows,ubuntu都可以。参考文章
需要anaconda,以及nvidia的驱动。如果你是ubuntu的话,可以参考这个文章装nvidia驱动。
1.创建环境
conda create -n your_env_name python=X.X
如:
conda create -n starcraft python=3.5
2.激活环境:
activate your_env_name
如:
activate starcraft
3.安装cuda和cudnn(这步可以省略,直接安装Tensorflow-gpu即可,会自动安装好依赖)
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu/anaconda/pkgs/free/linux-64/
conda install cudnn=7.1.4 -c https://mirrors.tuna.tsinghua.edu/anaconda/pkgs/main/linux-64/
4.安装tensorflow-gpu(这里需要和之前的版本对应到)。
pip install tensorflow-gpu==1.9.0
我的微信公众号名称:小小何先生
公众号介绍:主要研究强化学习、计算机视觉、深度学习、机器学习等相关内容,分享学习过程中的学习笔记和心得!期待您的关注,欢迎一起学习交流进步!
本文标签: 环境软件WindowsTensorFlowconda
版权声明:本文标题:软件类配置(一)【Windows下使用conda在虚拟环境中安装CUDA、CUDNN及Tensorflow】 内容由网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.betaflare.com/biancheng/1738021006a2050528.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论