Hacktoberfest 2026: los issues que los mantenedores marcaron para octubre, abiertos y aptos para principiantes. Explorar issues de Hacktoberfest

torchrec tutorial: old cuda version install

Abierto
#3,713 0 comentarios 0 reacciones 0 asignados Ver en GitHub

Los mantenedores suelen responder en 2 días

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
2/5
Tiempo estimado
1-3 horas
Aptitud para principiantes
48/100
Tipo de issue
Error
Claridad
Bien especificado
Estado de actividad
Estancado
Stack tecnológico
python, pytorch

Línea de trabajo

Comienza con intermediate_source/torchrec_intro_tutorial.py en las líneas 38-43 y reproduce la instalación en el entorno actual de Colab. Comprueba si los cambios propuestos en el paquete y el índice instalan dependencias de torchrec compatibles y ejecuta el tutorial lo suficiente para confirmar que la configuración se completa sin la incompatibilidad de versiones de CUDA.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

torchrec

https://github.com/pytorch/tutorials/blob/a3695df13baac449f92a06b1029260a21f2e5c90/intermediate_source/torchrec_intro_tutorial.py#L38C1-L43C3

the current lines try to install a cuda 12.1 version when the current colab is 12.6 (and so torch remains at 12.6)
changing the index to cu126 fails, because torchrec requires tensordict, which is not in the index?

my proposed solution is to remove the pytorch install line
and add tensordict to torchmetrics line

# !pip3 install --pre torch --index-url https://download.pytorch.org/whl/cu121 -U remove
!pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu126
!pip3 install torchmetrics==1.0.3 tensordict
!pip3 install torchrec --index-url https://download.pytorch.org/whl/cu126

cc @PaulZhang12 @iamzainhuda @vmoens @nairbv

Lenguaje dominante
Python
Estrellas
9.4k
Forks
4.4k
Merge medio
3 d 5 h
PR fusionados (30 d)
4

Preparar el entorno

Abrir en Codespaces

Inicia el contenedor de desarrollo del proyecto en tu navegador, con tu propia cuenta de GitHub.

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de pytorch/tutorials

Todos los issues de pytorch/tutorials

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.