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