La versión cuda de pytorch coincide, pero la operación aún informa un error

problema

Use nvidia-smila 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
Inserte la descripción de la imagen aquí

Supongo que te gusta

Origin blog.csdn.net/jokerxsy/article/details/107752374
Recomendado
Clasificación