Spaces:
Runtime error
Runtime error
File size: 771 Bytes
f670afc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
#!/bin/bash
key=${1}
rm -rf Dockerfile
echo "FROM nvcr.io/nvidia/pytorch:${key}-py3" > Dockerfile
input="Dockerfile.base"
while IFS= read -r line
do
echo "$line" >> Dockerfile
done < "$input"
input="scripts/requirements.txt"
while IFS= read -r line
do
echo "RUN pip install $line" >> Dockerfile
done < "$input"
for p in correlation channelnorm resample2d bias_act upfirdn2d; do
echo "COPY imaginaire/third_party/$p $p" >> Dockerfile
echo "RUN cd $p && rm -rf build dist *-info && python setup.py install" >> Dockerfile
done
# Compile GANcraft libraries.
echo "COPY imaginaire/model_utils/gancraft/voxlib gancraft/voxlib" >> Dockerfile
echo "RUN cd gancraft/voxlib && make" >> Dockerfile
docker build -t nvcr.io/nvidian/lpr-imagine/imaginaire:${key}-py3 .
|