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 --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple/