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的主要内容,如果未能解决你的问题,请参考以下文章

win10 + cuda10 +cudnn + GLUON 环境搭建

『MXNet』专题汇总

win10 配置tensorflow gpu环境

Win10 TensorFlow(gpu)安装详解

win10安装tensorflow-gpu

Win10+VScode +miniconda安装tensorflow-gpu