Pytorch 安装

nvcc -V 
nvidia-smi

看 CUDA 版本

上 https://pytorch.org/get-started/previous-versions/ 搜指令。

可以再加下面指令换源：

-i https://pypi.douban.com/simple/
-i https://pypi.tuna.tsinghua.edu.cn/simple/
-i https://mirrors.aliyun.com/pypi/simple/
-i https://pypi.org/simple/

比如：
pytorch==2.1.0+cu118:

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --extra-index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple/

pytorch 源特别慢，也可以换源成 aliyun（但是不太完整，有些版本没有）：

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0  -f https://mirrors.aliyun.com/pytorch-wheels/cu121/