Glenn Jocher
GitHub
pirms 3 gadiem
1 mainītis faili ar 3 papildinājumiem un 0 dzēšanām
+ 3
- 0
Dockerfile
Parādīt failu
| @@ -53,3 +53,6 @@ ENV HOME=/usr/src/app | |||
| # Update Ubuntu drivers | |||
| # https://www.maketecheasier.com/install-nvidia-drivers-ubuntu/ | |||
| # DDP test | |||
| # python -m torch.distributed.run --nproc_per_node 2 --master_port 1 train.py --epochs 3 | |||
Notiek ielāde…