MingGatsby's picture
Update requirements.txt
da773e7
raw
history blame
269 Bytes
torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118
torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118
torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
monai==1.2.0
opencv-python==4.8.0.76
timm==0.5.4
grad-cam==1.4.8