瀏覽代碼
Update Dockerfile (#4862)
modifyDataloader
Glenn Jocher
GitHub
3 年之前
沒有發現已知的金鑰在資料庫的簽署中
GPG Key ID: 4AEE18F83AFDEB23
共有
1 個文件被更改，包括
3 次插入 和
0 次删除
-
Dockerfile
|
|
|
@@ -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 |
