windows yolo11 删除cpu环境,安装GPU环境

时间:2024-11-12 22:39:24

一、查看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!