Open
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
Metadata
Metadata
Assignees
Labels
No labels