File tree Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -97,7 +97,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
97
97
sudo pip -q install awscli==1.16.35
98
98
99
99
if [ -n "${CUDA_VERSION}" ]; then
100
- DRIVER_FN="NVIDIA-Linux-x86_64-410.104 .run"
100
+ DRIVER_FN="NVIDIA-Linux-x86_64-430.40 .run"
101
101
wget "https://s3.amazonaws.com/ossci-linux/nvidia_driver/$DRIVER_FN"
102
102
sudo /bin/bash "$DRIVER_FN" -s --no-drm || (sudo cat /var/log/nvidia-installer.log && false)
103
103
nvidia-smi
Original file line number Diff line number Diff line change @@ -175,6 +175,11 @@ Deploying PyTorch Models in Production
175
175
Parallel and Distributed Training
176
176
---------------------------------
177
177
178
+ .. customgalleryitem ::
179
+ :tooltip: Model parallel training on multiple GPUs
180
+ :description: :doc: `/intermediate/model_parallel_tutorial `
181
+ :figure: _static/img/distributed/DistPyTorch.jpg
182
+
178
183
.. customgalleryitem ::
179
184
:tooltip: Getting started with DistributedDataParallel
180
185
:description: :doc: `/intermediate/ddp_tutorial `
@@ -318,6 +323,7 @@ PyTorch Fundamentals In-Depth
318
323
:hidden:
319
324
:caption: Parallel and Distributed Training
320
325
326
+ intermediate/model_parallel_tutorial
321
327
intermediate/ddp_tutorial
322
328
intermediate/dist_tuto
323
329
beginner/aws_distributed_training_tutorial
You can’t perform that action at this time.
0 commit comments