Skip to content

Commit 6f20c08

Browse files
Merge pull request NVIDIA#735 from NVIDIA/pribalta-fix-unetind-readme
[UNetInd] Add --gpus flag to docker run
2 parents 475256f + 81e8636 commit 6f20c08

File tree

1 file changed

+1
-1
lines changed
  • TensorFlow/Segmentation/UNet_Industrial

1 file changed

+1
-1
lines changed

TensorFlow/Segmentation/UNet_Industrial/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ For the specifics concerning training and inference, see the [Advanced](#advance
220220
# make a directory for results, for example ./results
221221
mkdir <path/to/results/directory>
222222
# start the container with nvidia-docker
223-
nvidia-docker run -it --rm \
223+
nvidia-docker run -it --rm --gpus all \
224224
--shm-size=2g --ulimit memlock=-1 --ulimit stack=67108864 \
225225
-v <path/to/dataset/directory>:/data/ \
226226
-v <path/to/result/directory>:/results \

0 commit comments

Comments
 (0)