分享缩略图

分享到:
链接已复制
首页> 新闻中心>

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 的问题。

【责任编辑:新华网】
返回顶部