Correr modelo que envió Tomás
Opened this issue · 0 comments
tabitaCatalan commented
Yeniffer se comunicó con Thomas, y él envió un modelo entrenado (el mail dice que es TF2, pero el código dice que es PyTorch).
CUDA 11.7 está disponible en el cluster. Para instalar Pytorch seguí las instrucciones de instalación para Linux usando Conda. Eliminé torchvision
y torchaudio
, y creé un ambiente lapnetv2
.
CONDA_OVERRIDE_CUDA="11.7" conda create --name lapnetv2
CONDA_OVERRIDE_CUDA="11.7" conda install pytorch pytorch-cuda=11.7 -c pytorch -c nvidia
CONDA_OVERRIDE_CUDA="11.7" conda install numpy matplotlib pyyaml h5py pandas json