resshift / docker /docker_build.sh
yuhj95's picture
Upload folder using huggingface_hub
4730cdc verified
#!/bin/bash
# Build the Docker image
# build an image with PyTorch 1.9, CUDA 11.1
# If you prefer other versions, just modified the Dockerfile
docker build -t resshift ../ --file ./Dockerfile