On Ubuntu 20.04 with driver 460 and Cuda 11.2, current docker compose did not work.
Until I added Gpu capability in docker-compose.yaml, pytorch did not see video card and complained about missing nvidia driver: RuntimeError: Found no NVIDIA driver on your system.
Next, after docker-compose.yaml fix, default version of torch, gave me this error:
GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
In case others have similar issues.
On Ubuntu 20.04 with driver 460 and Cuda 11.2, current docker compose did not work.
Until I added Gpu capability in docker-compose.yaml, pytorch did not see video card and complained about missing nvidia driver:
RuntimeError: Found no NVIDIA driver on your system.
Next, after docker-compose.yaml fix, default version of torch, gave me this error:
GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
These are changes I made to fix errors above:
You need to use docker-compose version
1.27
or newer.