Skip to content

Commit

Permalink
Update README.md with 0.10 Docker image name
Browse files Browse the repository at this point in the history
  • Loading branch information
tomaroberts committed Nov 12, 2021
1 parent b69d5d4 commit e34261f
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,12 +58,12 @@ git clone --recurse-submodules https://github.com/SVRTK/svrtk-docker-gpu.git /ho
### Option 1 --- Install using pre-built Docker image
#### 1.1 Download the pre-built Docker image from DockerHub:
```
docker pull fetalsvrtk/svrtk:svrtk-docker-gpu
docker pull fetalsvrtk/svrtk:svrtk-docker-gpu-0.10
```
#### 1.2 Test the Docker container is working:

```
docker run -it fetalsvrtk/svrtk:svrtk-docker-gpu
docker run -it fetalsvrtk/svrtk:svrtk-docker-gpu-0.10
```
Once inside the container, test MIRTK is working by running:

Expand Down Expand Up @@ -96,13 +96,13 @@ wget https://gin.g-node.org/SVRTK/fetal_mri_network_weights/raw/master/checkpoin
#### 2.3 Build the container using the Dockerfile:

```
docker build -f svrtk-docker-gpu.Dockerfile -t svrtk-docker-gpu .
docker build -f svrtk-docker-gpu.Dockerfile -t fetalsvrtk/svrtk:svrtk-docker-gpu-0.10 .
```

#### 2.4 Test the Docker container is working:

```
docker run -it svrtk-docker-gpu
docker run -it fetalsvrtk/svrtk:svrtk-docker-gpu-0.10
```
Once inside the container, test MIRTK is working by running:

Expand All @@ -122,7 +122,7 @@ Copy nifti files (labelled `stack1.nii.gz`, `stack2.nii.gz`, ... `stackN.nii.gz`
Then:

```
docker run -v "svrtk-docker-gpu/recon":/home/recon svrtk-docker-gpu /home/scripts/docker-recon-brain-auto.bash /home/recon
docker run -v "svrtk-docker-gpu/recon":/home/recon fetalsvrtk/svrtk:svrtk-docker-gpu-0.10 /home/scripts/docker-recon-brain-auto.bash /home/recon
```

This will output a 3D SVR-reconstructed volume named `SVR-output.nii.gz`.
Expand All @@ -131,5 +131,5 @@ This will output a 3D SVR-reconstructed volume named `SVR-output.nii.gz`.
##### Optional:
If you have successfully installed CUDA, you can run the GPU-accelerated version with:
```
docker run -it --gpus all -v "svrtk-docker-gpu/recon":/home/recon svrtk-docker-gpu /home/scripts/docker-recon-brain-auto.bash /home/recon
docker run -it --gpus all -v "svrtk-docker-gpu/recon":/home/recon fetalsvrtk/svrtk:svrtk-docker-gpu-0.10 /home/scripts/docker-recon-brain-auto.bash /home/recon
```

0 comments on commit e34261f

Please sign in to comment.