[Registro] Inconsistencia de la versión de visualización de nvcc y nvidia-smi

Cuando traté de ejecutar el programa con 3090 en el servidor, encontré un error.
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.
Verifiqué la versión de CUDA, y la versión nvcc -Vque obtuve fue V11.6.124, y nvidia-smila versión que obtuve fue 11.4.

El papel de nvcc y nvidia-smi

1. nvcc es un compilador CUDA que compila programas en archivos binarios ejecutables.
2. nvidia-smi ayuda a administrar y monitorear los dispositivos NVIDIA GPU.
   nvcc muestra la API de tiempo de ejecución de CUDA, que se instala mediante el instalador de CUDA Toolkit. nvidia-smi muestra la API del controlador.

selección de versión de pytorch

Al seleccionar la versión de pytorch, el comando correspondiente a la versión de CUDA especificada es cudatoolkit, por lo que se debe seleccionar el número de versión de nvcc -V.

cómo instalar

https://pytorch.org/get-started/locally/Si
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
elige usar la fuente del espejo doméstico, debe eliminar -c python -c nvidia
la fuente del espejo doméstico. Parece que no hay una adaptación de la versión 11.6, por lo que no puede usar el fuente de espejo doméstico al instalar una versión superior. Es un poco lento. Espera. Muy bien.

confirmar la instalación

Una vez completada la instalación, ejecute

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

---

referencia:
https://www.jianshu.com/p/eb5335708f2a
https://pytorch.org/get-started/locally/