Commit 5e6e34ed authored by chenpangpang's avatar chenpangpang
Browse files

Merge branch 'tensorflow' of...

Merge branch 'tensorflow' of http://developer.hpccube.com/codes/chenpangpang/gpu-base-image-build into tensorflow

Conflicts:
	script/3_image_test.sh
parents 32b6c77b aa879d4a
...@@ -9,7 +9,7 @@ fi ...@@ -9,7 +9,7 @@ fi
if [[ "$1" == *"pytorch"* ]]; then if [[ "$1" == *"pytorch"* ]]; then
docker run --rm --platform=linux/amd64 --gpus all -v ./$TARGET_DIR:/workspace --workdir /workspace/pytorch/stable-diffusion-v1-4 $1 python infer.py; fi docker run --rm --platform=linux/amd64 --gpus all -v ./$TARGET_DIR:/workspace --workdir /workspace/pytorch/stable-diffusion-v1-4 $1 python infer.py; fi
if [[ "$1" == *"tensorflow"* ]]; then if [[ "$1" == *"tensorflow"* ]]; then
tensorflow_version=$(echo "$1" | cut -d: -f2 | cut -d- -f1) tensorflow_version=$(echo "$1" | cut -d: -f2 | cut -d- -f1)
# 当tensorflow版本为2.16.1时,不添加环境变量找不到cuda,所以需要这样执行验证。在正常交互式启动容器时,会默认激活/etc/bash.bashrc,可以正常找到cuda # 当tensorflow版本为2.16.1时,不添加环境变量找不到cuda,所以需要这样执行验证。在正常交互式启动容器时,会默认激活/etc/bash.bashrc,可以正常找到cuda
if [[ "$tensorflow_version" == "2.16.1" ]]; then if [[ "$tensorflow_version" == "2.16.1" ]]; then
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment