docker/docker_build.sh

#!/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