torch、torchvision、torchaudio安装相应的cuda版本(cuda 12.1情况下)
2025-06-24 12:03:53
来源:新华网
首先进入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 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版本推理速度慢。如下图所示,#xff1a;安装cuda版本的包会更快。
您已经默认配置了清华镜像或阿里镜像源等。
PS: 同时解决torch问题.cuda.is_available=False 的问题。