[Запись] Несоответствие версии дисплея nvcc и nvidia-smi

Когда я попытался запустить программу с 3090 на сервере, я обнаружил ошибку.Я
UserWarning: NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
проверил версию CUDA, и nvcc -Vу меня была версия V11.6.124, а nvidia-smiу меня была версия 11.4.

Роль nvcc и nvidia-smi

1. nvcc — это компилятор CUDA, который компилирует программы в исполняемые двоичные файлы.
2. nvidia-smi помогает управлять и контролировать устройства NVIDIA GPU.
   nvcc показывает API среды выполнения CUDA, который устанавливается установщиком CUDA Toolkit. nvidia-smi показывает API драйвера.

выбор версии pytorch

При выборе версии pytorch команда, соответствующая указанной версии CUDA, — cudatoolkit, поэтому следует выбрать номер версии nvcc -V.

Как установить

https://pytorch.org/get-started/locally/Если
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
вы решите использовать внутренний зеркальный источник, вам необходимо удалить -c python -c nvidia
внутренний зеркальный источник.Похоже, что адаптации версии 11.6 нет, поэтому вы не можете использовать Внутренний зеркальный источник при установке более высокой версии.Это немного медленно.Подождите.Все нормально.

подтвердить установку

После завершения установки запустите

import torch
x = torch.rand(5,3)
print(x)
print(torch.cuda.is_available())

---

ссылка:
https://www.jianshu.com/p/eb5335708f2a
https://pytorch.org/get-started/locally/