PyTorch NGC Container for WSL 2

Hello,
when I try to run nvcr.io/nvidia/pytorch:20.07-py3 on WSL2 I get the following warning which prevents to image from running:

WARNING: The NVIDIA Driver was not detected. GPU functionality will not be available.
Use ‘nvidia-docker run’ to start this container; see
GitHub - NVIDIA/nvidia-docker: Build and run Docker containers leveraging NVIDIA GPUs .

NOTE: MOFED driver for multi-node communication was not detected.
Multi-node communication performance may be reduced.

NOTE: The SHMEM allocation limit is set to the default of 64MB. This may be
insufficient for PyTorch. NVIDIA recommends the use of the following flags:
nvidia-docker run --ipc=host …

Please advise. Thanks!

1 Like