Quellcode durchsuchen

Update Dockerfile (#4862)

modifyDataloader
Glenn Jocher GitHub vor 3 Jahren
Ursprung
Commit
2809616345
Es konnte kein GPG-Schlüssel zu dieser Signatur gefunden werden GPG-Schlüssel-ID: 4AEE18F83AFDEB23
1 geänderte Dateien mit 3 neuen und 0 gelöschten Zeilen
  1. +3
    -0
      Dockerfile

+ 3
- 0
Dockerfile Datei anzeigen

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

Laden…
Abbrechen
Speichern