<Ubuntu 20.04 LTS> Как установить Cuda для torch 1.8.1 [закрыто]
Я использую ubuntu 20.04 LTS в Windows10. И я собираюсь использовать свой графический процессор в качестве фонарика.
Что мне установить?
cuda | cuDNN | cuda driver
На самом деле, я не знаю, нужен ли cuDNN.
Ниже приведены технические характеристики моего компьютера.
- Графический процессор: NVIDIA GEFORCE GTX 1050
- Framework: torch 1.8.1 (это последняя версия 1.8.1+ cu111?)
- ОС: Ubuntu 20.04 LTS
Я много пробовал, но, возможно, версия не правильная.
Мои основные вопросы следующие.
- Должен ли я установить cuda, cuDNN, все драйверы?
- Какие версии мне установить?