Open sizhit2 opened 2 days ago
Add device option in docker_build_dependency_image.sh. The default option is still Tpu. To build docker image with GPU, just specify bash docker_build_dependency_image.sh DEVICE=gpu Add maxdiffusion_gpu_dependencies.Dockerfile .
docker_build_dependency_image.sh
bash docker_build_dependency_image.sh DEVICE=gpu
maxdiffusion_gpu_dependencies.Dockerfile
Add device option in
docker_build_dependency_image.sh
. The default option is still Tpu. To build docker image with GPU, just specifybash docker_build_dependency_image.sh DEVICE=gpu
Addmaxdiffusion_gpu_dependencies.Dockerfile
.