Skip to content

Commit

Permalink
Pin torch to be below 1.13
Browse files Browse the repository at this point in the history
Torch 1.13 has some weird extra NVIDIA dependencies, described [here] (pytorch/pytorch#88049), which don't seem to recognize that cuda is loaded as a module. So as it currently stands, you need to install top-bottom and then uninstall those nvidia packages, and then maybe muck around a bit more so that everything points to the right place. I don't see any issue with pinning torch 1.12 for now.
  • Loading branch information
jonahpearl committed Feb 6, 2023
1 parent e57b1ff commit 338a8ae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
Expand Up @@ -36,7 +36,7 @@
'scikit-learn',
'scipy',
'segmentation_models_pytorch',
'torch',
'torch<1.13',
'torchinfo',
'torchplot',
'torchvision',
Expand Down

0 comments on commit 338a8ae

Please sign in to comment.