Как добавить драйверы cuda в gcp Ubuntu vm?

Question

Когда я запускаю модель pytorch на виртуальной машине Google, используя:

model.cuda()

, я получаю эту ошибку:

AssertionError: 
Found no NVIDIA driver on your system. Please check that you
have an NVIDIA GPU and installed a driver from
http://www.nvidia.com/Download/index.aspx

Answer 1

Начиная с 2020-05-01, машины GCP по умолчанию работают с 9.12. Поэтому трудно использовать репозиторий по умолчанию:

ppa:graphics-drivers

Я рекомендую просто загрузить драйвер и самостоятельно установить его:

wget http://us.download.nvidia.com/XFree86/Linux-x86_64/440.82/NVIDIA-Linux-x86_64-440.82.run

Затем запустить его в режиме sudo:

 sudo bash NVIDIA-Linux-x86_64-440.82.run

Примите все, и вы можете использовать свой графический процессор