torch、torchvision、torchaudio安装对应cuda版本（cuda 12.1情况下）

首先进入anaconda prompt

然后切换到你的虚拟环境 conda activate 虚拟环境名称。如： conda activate yolov8

之后pip安装对应包。不要使用conda install ！！！

对应代码：

pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121

如果安装cuda 11.8版本的，把最后的cu121改成cu118即可。目前支持最高cuda 12.2。

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

因为网上 pip install torch 等全部默认安装cpu版本，推理速度慢。如下图所示：

安装cuda版本的包会更快。默认你已经配置了清华镜像或阿里镜像源等。

PS: 同时解决了torch.cuda.is_available=False 的问题。