基于Anaconda3环境下的CNTK安装

安装环境:

  1. operating system: Ubuntu16.04
  2. cuda version: 9.1
  3. cuDNN version: 7.0
  4. python version: 3.5
  5. CNTK version: 2.4

为了避免冲突,我们选择把CNTK安装在虚拟环境中

可以选择python的pyenv 和 virtualenv 或者 anaconda自带的conda。这里我使用的是anaconda3自带的conda

首先到anaconda官网,下载anaconda3
anaconda下载完后,文件应该是如下图所示

接着,运行下面的命令安装anaconda

$ sh Anaconda3-5.1.0-Linux-x86_64.sh

anaconda的安装很简单,这里就不多描述。

CNTK需要你的系统安装有OpenMPI。在Ubuntu中可以通过以下命令安装

$ sudo apt install openmpi-bin

然后,创建名为cntk-py35的虚拟环境

$ conda create --name cntk-py35 python=3.5 numpy scipy h5py jupyter

激活cntk虚拟环境

$ source activate cntk-py35

关闭cntk虚拟环境

$ source deactivate

激活虚拟环境后,用pip安装CNTK(GPU)即可

$ pip install https://cntk.ai/PythonWheel/GPU/cntk-2.4-cp35-cp35m-linux_x86_64.whl

测试CNTK是否安装成功并输出CNTK版本

$ python -c "import cntk; print(cntk.__version__)"

CNTK测试样例运行截图:

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值