File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
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 @@ -165,7 +165,7 @@ Parallel and Distributed Training
165
165
---------------------------------
166
166
167
167
.. customgalleryitem ::
168
- :tooltip: Train large models with multiple GPUs using model parallel
168
+ :tooltip: Model parallel training on multiple GPUs
169
169
:description: :doc: `/intermediate/model_parallel_tutorial `
170
170
:figure: _static/img/distributed/DistPyTorch.jpg
171
171
You can’t perform that action at this time.
0 commit comments