Skip to content

Commit d7f2929

Browse files
committed
Update timm installation instructions
1 parent 3834142 commit d7f2929

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

index.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https
157157
pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cpu
158158

159159
# Finally install some packages that rely on the Pytorch installation
160-
pip install timm pytorch-msssim torchmetrics[image]==1.4.*
160+
pip install timm==1.0.14 pytorch-msssim torchmetrics[image]==1.4.*
161161
```
162162

163163
The PyPI package does not install <a href="https://pytorch.org/get-started/locally/" target="_blank" rel="noopener noreferrer">Pytorch</a> because there is no option to build that package specifying exactly the CUDA version you want to use. There are a few solutions to set up ``pyproject.toml`` with poetry and specify the CUDA version, as discussed <a href="https://github.com/python-poetry/poetry/issues/6409" target="_blank" rel="noopener noreferrer">here</a>, but then PyPI package can not be built (as stated <a href="https://peps.python.org/pep-0440/#direct-references" target="_blank" rel="noopener noreferrer">here</a>).
@@ -198,7 +198,7 @@ mamba install pytorch::pytorch torchvision torchaudio -c pytorch
198198
Then, add extra pytorch related packages:
199199

200200
```bash
201-
mamba install timm torchmetrics
201+
mamba install timm==1.0.14 torchmetrics
202202
```
203203

204204
Install BiaPy Dependencies:
@@ -254,7 +254,7 @@ pip install --editable .
254254

255255
# Install Pytorch and GPU dependencies
256256
pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118
257-
pip install timm pytorch-msssim torchmetrics[image]==1.4.*
257+
pip install timm==1.0.14 pytorch-msssim torchmetrics[image]==1.4.*
258258
```
259259
{% endtab %}
260260

@@ -265,7 +265,7 @@ pip install --editable .
265265

266266
# Install Pytorch and GPU dependencies
267267
pip install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
268-
pip install timm pytorch-msssim torchmetrics[image]==1.4.*
268+
pip install timm==1.0.14 pytorch-msssim torchmetrics[image]==1.4.*
269269
```
270270
{% endtab %}
271271
<!-- command_line_CUDA_installation -->
@@ -398,4 +398,4 @@ if (OSName === "linux") {
398398
} else {
399399
elem3.setAttribute('href', "/docker_inst/#mac_install");
400400
}
401-
</script>
401+
</script>

0 commit comments

Comments
 (0)