0


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


本文转载自: https://blog.csdn.net/qq_61661957/article/details/140160823
版权归原作者 YourDickie 所有, 如有侵权,请联系我们删除。

“torch、torchvision、torchaudio安装对应cuda版本(cuda 12.1情况下)”的评论:

还没有评论