afdaniele/x-docker

use NVIDIA_DRIVER_CAPABILITIES=all instead of NVIDIA_DRIVER_CAPABILITIES=graphics?

Closed this issue · 1 comments

cbschaff commented

I can't run pytorch code on a gpu unless more driver capabilities are loaded. Is there a reason you are only using graphics? If not, can you change this to flag to "all" instead?

afdaniele commented

Fixed in 00e3306