喜大普奔!!! 使用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.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.1.4 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

4.安装tensorflow-gpu(这里需要和之前的版本对应到)。

pip install tensorflow-gpu==1.9.0

我的微信公众号名称:小小何先生
公众号介绍:主要研究强化学习、计算机视觉、深度学习、机器学习等相关内容,分享学习过程中的学习笔记和心得!期待您的关注,欢迎一起学习交流进步!

GitHub 加速计划 / te / tensorflow
21
4
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:1 个月前 )
4f64a3d5 Instead, check for this case in `ResolveUsers` and `ResolveOperand`, by querying whether the `fused_expression_root` is part of the `HloFusionAdaptor`. This prevents us from stepping into nested fusions. PiperOrigin-RevId: 724311958 1 个月前
aa7e952e Fix a bug in handling negative strides, and add a test case that exposes it. We can have negative strides that are not just -1, e.g. with a combining reshape. PiperOrigin-RevId: 724293790 1 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐