win10 + gluon + GPU
Posted 学渣
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了win10 + gluon + GPU相关的知识,希望对你有一定的参考价值。
1. 下载教程
可以用浏览器下载zip格式并解压,在解压目录文件资源管理器的地址栏输入cmd
进入命令行模式。
也可以
git pull https://github.com/mli/gluon-tutorials-zh
2.安装gluon CPU
添加源:
# 优先使用清华conda镜像 conda config --prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ # 也可选用科大conda镜像 conda config --prepend channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
cmd中安装
conda env create -f environment.yml activate gluon # 注意Windows下不需要 source
可更新教程:
conda env update -f environment.yml
3.安装GPU版本
先卸载CPU
pip uninstall mxnet
然后
pip install --pre mxnet-cu75 # CUDA 7.5 pip install --pre mxnet-cu80 # CUDA 8.0
【可选项】国内用户可使用豆瓣pypi镜像加速下载:
pip install --pre mxnet-cu75 -i https://pypi.douban.com/simple # CUDA 7.5 pip install --pre mxnet-cu80 -i https://pypi.douban.com/simple # CUDA 8.0
以上是关于win10 + gluon + GPU的主要内容,如果未能解决你的问题,请参考以下文章