Skip to content

Commit 2ff9485

Browse files
wkentarosoumith
authored andcommitted
Add instruction to use specified GPU id (#73)
Related to #72 Is this information available in docs?
1 parent 190f8fa commit 2ff9485

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

mnist/README.md

+1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,5 @@
33
```bash
44
pip install -r requirements.txt
55
python main.py
6+
# CUDA_VISIBLE_DEVICES=2 python main.py # to specify GPU id to ex. 2
67
```

0 commit comments

Comments
 (0)