English
medical
brain-data
mri
brain2vec / requirements.txt
jesseab's picture
Updated requirements
2bbcdec
raw
history blame contribute delete
431 Bytes
# requirements.txt
# PyTorch (CUDA or CPU version).
torch>=1.12
# Install MONAI Generative first
monai-generative
# Now explicitly require the newer MONAI version.
monai>=1.2
# Now force reinstall MONAI Weekly so its (newer) MONAI version takes precedence
monai-weekly
# For perceptual losses in MONAI's generative module.
lpips
# Common Python libraries
pandas
numpy
nibabel
tqdm
tensorboard
matplotlib
datasets
scikit-learn