27 lines
733 B
Bash
27 lines
733 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Set variables
|
||
|
|
IMAGE_NAME="prebuilts-rocm:6.1.2"
|
||
|
|
CONTAINER_NAME="prebuilts-rocm"
|
||
|
|
FILES_TO_COPY=["/tmp/bitsandbytes/dist/", "/tmp/llama-cpp-python/dist/"]
|
||
|
|
WHERE_TO_PASTE="./build_output/"
|
||
|
|
|
||
|
|
# Run the Docker container
|
||
|
|
docker run -d --name $CONTAINER_NAME $IMAGE_NAME
|
||
|
|
|
||
|
|
# Check if the container is running
|
||
|
|
if [ "$(docker ps -q -f name=$CONTAINER_NAME)" ]; then
|
||
|
|
echo "Container $CONTAINER_NAME is running."
|
||
|
|
|
||
|
|
# Copy the files from the container to the host
|
||
|
|
for file in $FILES_TO_COPY; do
|
||
|
|
docker cp $CONTAINER_NAME:$file $WHERE_TO_PASTE
|
||
|
|
done
|
||
|
|
|
||
|
|
echo "Files copied to $WHERE_TO_PASTE."
|
||
|
|
else
|
||
|
|
echo "Failed to start container $CONTAINER_NAME."
|
||
|
|
fi
|
||
|
|
|
||
|
|
docker stop $CONTAINER_NAME
|
||
|
|
docker rm $CONTAINER_NAME
|