problema
Use nvidia-smi
la versión cuda del servidor de vistas y encontré que es 10.1, así que fui al sitio web oficial para descargar el pytorch (1.6.0) correspondiente a la versión cuda. Sin embargo, cuando ejecuté el código, todavía informé un error que decía que la versión no coincidía
resolver
Descargue la versión 1.3.0 de pytorch (correspondiente a la versión 10.0 de cuda) y luego
use con éxito el comando:pip install --user torch==1.3.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html