[Tools] PyTorch installation (matching CUDA version)

CUDA Toolkit Archive (including all versions of PyTorch)

Find the corresponding download command according to the CUDA version and PyTorch version you need. Here we take PyTorch1.10.1 CUDA 11.3 as an example.

insert image description here

Linux and Windows

# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cpuonly -c pytorch

Guess you like

Origin blog.csdn.net/guai7guai11/article/details/131060347