- 对于2.0版本往上的pytorch python版本必须大于等于3.8
- 对于小于2.0版本的pytorch 下载GPU版本时必须要加上
--extra-index-url https://download.pytorch.org/whl/cu117
其中 cu117为你cuda的版本,例如
# CUDA 11.7
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117