Это первый день моего ноябрьского обновления
1.UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010)
анализировать:
Причина ошибки в том, что версия CUDA слишком низкая, то есть версия CUDA
- Повысить версию CUDA
- версия с нижней горелкой
Проверять:
print(torch.version.cuda)
Глядя на версию cuda, используемую факелом, результат равен 10.2.
nvcc -V
Проверьте версию cuda текущей видеокарты и обнаружите, что это 10.1.
На данный момент мы можем знать, что ошибка вызвана несоответствием между графической картой и версией cuda факела.
решать:
Нажмите на следующую ссылку, чтобы войти на официальный сайт pytorch и найти версию, поддерживаемую cuda10.1.py torch.org/individual-started…Выберите torch1.7.1, чтобы загрузить и установить версию, соответствующую cuda10.1.Поскольку conda не поддерживается в Китае, выберите pip для установки.
# CUDA 10.1
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
проверять:
Запускаем программу еще раз и ошибка пропадает, torch.version.cuda тоже меняется обратно на 10.1, проблема решена! !