search

NVIDIA / ansible-role-nvidia-driver

BSD 3-Clause "New" or "Revised" License
117 stars 67 forks source link

How do i install 455 drivers for CUDA 11.1 #34

Closed ghost closed 3 years ago

ghost commented 3 years ago

HI, If i set nvidia_driver_ubuntu_branch: 455 i get No package matching 'nvidia-headless-455-server' is available? (should this be trying to install something that doesn't exist?)

If i try also setting:

nvidia_driver_ubuntu_packages:
- nvidia-headless-450-server
- nvidia-utils-450-server

and 450 is installed not 455? I'm running this on k3s on docker and using ansible-role-nvidia-docker to setup that up. I need Nvidia's Triton Infernece Server 20.11 which needs CUDA 11.1 and thus minimum 455 drivers. I can't use 460 because the Nvidia k8s-device-plugin breaks.

So i'm stuck between a rock and a hard place. If i could get 455 drivers working here then I'd be OK (for now). Is this possible with this role?

ghost commented 3 years ago

I think i got it:

   vars:
        nvidia_driver_ubuntu_install_from_cuda_repo: yes
        nvidia_driver_package_version: "465.19.01-1"