红联Linux门户
Linux帮助

Ubuntu配置python环境和mxnet

发布时间:2016-03-18 15:39:36来源:linux网站作者:panda1942

1.python 环境

我只安装Python 2.7,没装virtualenv,虚拟环境 mxnet 提示缺少 numpy,懒得搞了。


2.mxnet

参考 phunter.farbox.com/post/mxnet-tutorial1

1.mxnet 基本依赖包

sudo apt-get update
sudo apt-get install -y build-essential git libatlas-base-dev libopencv-dev

2.下载 mxnet

git clone --recursive https://github.com/dmlc/mxnet

3.安装CUDA

https://developer.nvidia.com/cuda-downloads

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.5-18_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb
sudo apt-get update
sudo apt-get install cuda

安装成功,可以用nvidia-smi命令查看你的显卡使用情况

4.编译GPU版 mxnet

复制 /mxnet/make 下 文件 config.mk 到 /mxnet 下

修改  USE_CUDA = 1
USE_CUDA_PATH = /usr/local/cuda

其他选项也可修改, 如 cudnn。

make -j4

5.可能要重新配置一下python

sudo apt-get install python-numpy

6.例子

python example/image-classification/train_mnist.py --network lenet --gpus 0

如果出现 ImportError: libcudart.so.7.0: cannot open shared object file: No such file

把CUDA的动态链接库加入PATH里, 在 root 目录下  bashrc 文件 里面加入

export LD_LIBRARY_PATH=/usr/local/cuda-7.5/targets/x86_64-linux/lib/:$LD_LIBRARY_PATH


本文永久更新地址:http://www.linuxdiyf.com/linux/19047.html