Skip to content

Incorrect install command for torch 1.8.1 using conda #925

Open
@zookae

Description

@zookae

📚 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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions