一、查看cuda版本
在cmd 输入nvidia-smi
得知cuda版本是12.4
二、删除 torch:
pip uninstall torch torchaudio torchvision
三、安装gpu环境:
拿到 url 后使用迅雷下载到桌面。进入桌面:
cd C:\Users\76211\Desktop
进入环境:
conda activate yolo
如图:
安装torch:
pip install "torch-2.4.1+cu124-cp38-cp38-win_amd64.whl"
输入
pip list
说明是gpu版本。然后安装torchvision torchaudio
pip3 install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
再次输入
pip list
OK!