目录
1、本地安装方法(1)
前提是已安装CUDA和Caffe等深度学习框架
1.1 直接apt安装:
sudo apt-get update
sudo apt-get install digits
容易因为各种原因导致安装不上,多试几次。
1.2 运行digits
python -m digits
注意:digits用的是python2
2、本地安装方法(2)
前提也是已安装CUDA和Caffe等深度学习框架
这种方法可以安装最新版的digits,目前是6.1.1.
2.1 下载digits
cd ~
git clone https://github.com/NVIDIA/DIGITS.git digits
2.2 安装依赖
cd digits
#若使用linux系统环境下的python,请使用该语句
sudo apt-get install graphviz gunicorn for req in $(cat requirements.txt); do sudo pip install $req; done
#若使用Anaconda环境下的python,使用该语句进行安装依赖库
pip install -r ~/digits/requirements.txt
2.3 运行digits
./digits-devserver
3、下载官方digits的docker镜像
3.1 拉取镜像
docker pull nvidia/digits
3.2 创建容器
nvidia-docker run -it --privileged=true -p 5000:5000 -v /home/digits/:/home/digits nvidia/digits:6.0
3.3 运行digits
进入容器:
docker exec -it XXXXXXXX bash ##XXXXXX为容器ID
运行digits:
python -m digits
4、digits服务使用
digits默认使用的是5000端口,注意不要和其它服务存在冲突,如果使用容器部署digits服务的话,要将容器的5000端口映射出来,建议也是映射到宿主机的5000端口,这时远程访问服务的5000端口就能打开digits页面。