基于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 或者 anaconda3自带的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测试样例运行截图:

猜你喜欢

转载自blog.csdn.net/Jonms/article/details/79550512
今日推荐