# DTK Container Toolkit ## 简介 DTK Container Toolkit 使用户能够构建和运行使用DCU设备的容器,该toolkit包括以下工具包。 ```dtk-container-toolkit``` - DTK容器运行时 ```dtk-ctk``` - DTK容器工具集命令行 ## 使用 首先确保已经安装好DTK。 ### 安装 使用 dpkg/rpm -i 进行安装。 重启docker服务 ```sh $ sudo systemctl restart docker ``` ### 在容器中使用HCU #### 通过 docker CLI 可以通过 docker run 添加参数 --gpus 给容器添加HCU设备。 ```sh $ docker run -it --gpus all ubuntu:18.04 # 添加所有HCU设备 $ docker run -it --gpus 1 ubuntu:18.04 # 添加一个HCU设备,HCU 0 ``` #### 通过环境变量 `DTK_VISIBLE_DEVICES` 可以通过 docker run 添加环境变量 -e DTK_VISIBLE_DEVICES 给容器添加HCU设备。 ```sh docker run -it -e DTK_VISIBLE_DEVICES=all ubuntu:18.04 # 添加所有HCU设备 docker run -it -e DTK_VISIBLE_DEVICES=0 ubuntu:18.04 # 添加HCU设备0 docker run -it -e DTK_VISIBLE_DEVICES=0,1 ubuntu:18.04 # 添加HCU设备0、1 ```