Add InfuseAI Image
Last updated
Last updated
InfuseAI provides and maintains a serious of public images on DockerHub. This document shows how to add images provided by InfuseAI.
First of all we check and pick up a pari of images from .
They are TensorFlow 2.1/CPU/Python3.7
and TensorFlow 2.1/GPU CUDA 10.1/Python3.7
.
We can learn URLs respectively from the list.
TensorFlow 2.1/CPU/Python 3.7: infuseai/docker-stacks:tensorflow-notebook-tf-v2-3f48358e
TensorFlow 2.1/GPU CUDA10.1/Python 3.7: infuseai/docker-stacks:tensorflow-notebook-3f48358e-gpu-cuda-10
Log in as a Group admin or as an Administrator.
Enter Image
Management from User Portal/Admin Portal respectively and click + Add
.
Fill in Name
: such as tf-21
.
Fill in Description
: such as TensorFlow 2.1 + Cuda10.1 + Python3.7
.
Select Type
: Universal
。
In terms of Universal
, we can assign a CPU-only image and a GPU-supported image respectively. When spawning a Notebook, PrimeHub will pick up the corresponding image according to the specified instance type (CPU-only or GPU-required).
Fill in Container image url
: CPU-only infuseai/docker-stacks:tensorflow-notebook-tf-v2-3f48358e
Check off Specific container image url for GPU
and fill in GPU-supportedinfuseai/docker-stacks:tensorflow-notebook-3f48358e-gpu-cuda-10
.
Because InfuseAI's registry on DockerHub is public, there is no need of Use Image Pull Secret
.
Global
or specify accessing Groups
.
Enter Notebook
from User Portal.
Select a group which can utilize the newly added image.
Select an instance type which requires small CPU resource. (just for verification)
Select the newly added Image tf-21
.
Start Notebook
. It takes time to pull off the newly added image for the first time; As long as a Notebook is launched successfully, the image is pulled successfully.
Verify GPU Image (Please avoid affecting users ):
Close/Stop previously launched Notebook.
Enter Notebook
from User Portal.
Select a group which can utilize the newly added image.
Select an instance type which requires small GPU resource. (just for verification)
Select the newly added Image tf-21
.
Start Notebook
. It takes time to pull off the newly added image for the first time; As long as a Notebook is launched successfully, the image is pulled successfully.