Skip to content

Incorrect install command for torch 1.8.1 using conda #925

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
zookae opened this issue Feb 4, 2022 · 1 comment
Open

Incorrect install command for torch 1.8.1 using conda #925

zookae opened this issue Feb 4, 2022 · 1 comment

Comments

@zookae
Copy link

zookae commented Feb 4, 2022

📚 Documentation

The "previous versions" installation guide indicates conda installs (for linux & windows) of torch 1.8.1 should use cudatoolkit=11.3; this should be cudatoolkit=11.1. The conda website does not have binaries for cuda 11.3 for torch 1.8.1, but does have cuda 11.1 binaries. This would also make sense since it matches the pip wheel instructions for torch 1.8.1 (and cudatoolkit for torch 1.8.0).

Link to the documentation:
https://pytorch.org/get-started/previous-versions/#linux-and-windows-2

conda binaries for comparison: https://anaconda.org/pytorch/pytorch/files?version=1.8.1&type=conda&page=4

I think the change just amounts to swapping from:

# CUDA 11.3
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge

to

# CUDA 11.1
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.1 -c pytorch -c conda-forge
@NeoKish
Copy link
Contributor

NeoKish commented Feb 9, 2022

The pull request #902 does fix that error

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants