安装环境:
- operating system: Ubuntu16.04
- cuda version: 9.1
- cuDNN version: 7.0
- python version: 3.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测试样例运行截图: