Hi @mrshenli
getting below error for mentioned example, how to disable cuda.
  File "/apps/dl/torch/ddp.py", line 20, in example
	model = nn.Linear(10, 10).to(rank)
	
	  File "/apps/miniconda3/envs/torch/lib/python3.8/site-packages/torch/cuda/__init__.py", line 94, in _check_driver
	raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

i haven’t installed CUDA and not required also…