首先进入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 的问题。
版权归原作者 YourDickie 所有, 如有侵权,请联系我们删除。