pytorch / pytorch/pytorch.github.io

Incorrect install command for torch 1.8.1 using conda

Open
#925 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
HTML
Stars
285
Forks
317
PR merge metrics
No merged PRs in 30d

Description

📚 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

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Open the linked previous-versions installation guide and compare the torch 1.8.1 conda commands with the available binaries linked in the issue. Change the Linux and Windows CUDA 11.3 examples to CUDA 11.1, then verify that the displayed package versions and commands match the issue’s proposed command.

Written by the indexing model from the issue text.

Assessment

Tech stack
anaconda, pytorch
Domain
documentation
Issue type
Documentation
Difficulty
1/5
Estimated time
Under an hour
Activity status
Stale
Clarity
Clearly specified
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.