Windows/Mac/Linux通过Docker安装计图
为了能够全平台支持计图的使用,计图研发团队提供了通过Docker安装计图的方法。通过Docker安装计图,无需用户准备过多的依赖,我们已经全部帮您打包到Docker镜像当中。
第一步:安装Docker
我们为您列举了Windows/Mac/Linux安装Docker的方法,您可以参阅我们的方法安装Docker,或者您也可以通过Docker官网查询安装方法。
目前,Docker支持的操作系统如下,请检查您的系统是否满足:
- Windows(>=10 专业版)
- Mac(>=10.13)
- Linux(CentOS, Debian, Fedora, Raspbian, Ubuntu)
如果您的操作系统不满足要求(比如Windows家庭版或者Windows 8),可以考虑安装Docker Toolbox,我们将在下面同时介绍Docker Toolbox的安装方法。
目前,Windows,Mac仅支持计图的CPU镜像,Linux同时支持CPU和CUDA,为了最优化体验,我们强烈推荐条件允许的同学安装原生Linux。
Windows 安装
点击此处下载Docker的Windows安装包, 双击运行Docker for Windows Installer
。
安装过程中可能需要重启,请遵循Docker安装程序的指示完成安装。
安装完成以后,右下角的托盘图标会显示Docker Logo小鲸鱼,表示Docker正在运行前,打开Docker的设置,在资源设置里,请把所有的CPU资源分配给Docker,并且分配至少4G的内存。
Windows 安装 Docker Toolbox
如果您的系统不满足安装Docker Desktop的条件,可以考虑安装Docker Toolbox,性能上Docker Toolbox会比Docker Desktop差。
您可以从这里下载DockerToolbox的Windows安装程序,遵循指令安装即可。
安装完成后,启动docker quick start terminal
, 可能会自动下载boot2docker
镜像,如果自动下载失败,您也可以从这里手动下载, 下载完成以后,将其放置在C:\Users\YourName\.docker.machine\cache\boot2docker.iso
(下图红色箭头对应路径)下。
安装成功后,会出现如下界面:
至此,您已经成功在windows上安装docker toolbox了,可以开始第二步了。
官方参考资料:https://docs.docker.com/toolbox/toolbox_install_windows/
Mac 安装
点击此处下载Docker的Mac安装包,双击运行Docker.dmg
。
安装完成以后,顶部状态栏会显示Docker已经在运行:
在运行前,打开Docker的设置,在资源设置里,请把所有的CPU资源分配给Docker,并且分配至少4G的内存。
Linux 安装
此处仅演示在Ubuntu(指令集x86-64)的安装方法,其他Linux发行版请参考官方的安装指南
ubuntu 20.04 安装方法:
sudo apt update
sudo apt install docker.io
sudo systemctl enable --now docker
ubuntu 16.04/18.04 安装方法:
# 删除旧版本的Docker
sudo apt-get remove docker docker-engine docker.io containerd runc
# 设置Docker仓库
sudo apt-get update
sudo apt-get install \
apt-transport-https \
ca-certificates \
curl \
gnupg-agent \
software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo apt-key fingerprint 0EBFCD88
sudo add-apt-repository \
"deb [arch=amd64] https://download.docker.com/linux/ubuntu \
$(lsb_release -cs) \
stable"
# 安装Docker
sudo apt-get update
sudo apt-get install docker-ce docker-ce-cli containerd.io
第二步:安装计图镜像
运行如下命令(Windows请使用PowerShell,Linux/Mac请使用Terminal):
docker run jittor/jittor python3.7 -m jittor.test.test_example
输出结果:
如果您的输出结果如上图所示,那么恭喜您,计图镜像已经安装成功了!
如果您在下载镜像时出现网络问题,也可以在这里下载jittor.tgz, 然后使用docker load -i ./jittor.tgz
手动加载镜像。
第三步:在浏览器打开计图笔记本
接下来,我们启动一个notebook server
,从命令行里面运行如下命令:
# Linux 用户
docker run -it --network host jittor/jittor
# Windows/Mac 用户
docker run -it -p 8888:8888 jittor/jittor
输出结果如下:
我们将输出的链接地址(http://127.0.0.1:8888/?token=xxxxx)复制到浏览器打开(Docker Toolbox用户请使用地址http://192.168.99.100:8888/?token=xxxx),会看到如下结果:
这里面罗列了很多计图提供的教程。我们打开LSGAN.cn.ipynb
,运行里面的教程代码试试吧!
运行的时候请留意内存限制,调整合适的batch_size
。
计图社区是开源开放的,期待您的建议和反馈。如果您喜欢计图,请star我们,或者参与贡献。
(可选)第四步:使用GPU进行加速
在docker中使用GPU加速jittor需要满足以下条件:
- 至少有一张nvidia显卡
- linux系统且安装好了nvidia-docker(docker版本>=19)
- CUDA (>=10.2)
动手能力比较强的同学,在满足了以上条件以后,可以使用如下命令运行jittor的cuda版本:
docker run --gpus all -it --network host jittor/jittor-cuda
如果您在下载镜像时出现网络问题,也可以在这里下载jittor-cuda.tgz, 然后使用docker load -i ./jittor-cuda.tgz
手动加载镜像。
如果您碰到任何困难,欢迎随时向我们反馈。
镜像下载
如下的镜像可以通过手动下载以后,使用docker load -i ...
加载:
参考资料
- Docker 官方安装教程: https://docs.docker.com/get-docker/
- Docker Desktop Installer 原始链接:https://download.docker.com/win/stable/Docker%20Desktop%20Installer.exe
- DockerToolbox 原始链接: https://github.com/docker/toolbox/releases/download/v19.03.1/DockerToolbox-19.03.1.exe
- boot2docker.iso 原始链接: https://github.com/boot2docker/boot2docker/releases/download/v19.03.5/boot2docker.iso
- Docker.dmg 原始链接: https://download.docker.com/mac/stable/Docker.dmg