查看显卡版本

此电脑-属性-设备管理器-显示适配器
image.png

如果装了nvidia驱动的话也可以用命令列出所有GPU

nvidia-smi -L

image.png

nvidia驱动

到官网,根据上面查到的显卡信息,输入后获得合适的驱动程序,下载安装。
image.png

nvidia-smi

image.png
ps:与较新的驱动程序相关联的nvidia-smi显示的CUDA版本是DRIVER API兼容性版本。它根本不表示任何有关实际安装的CUDA版本的信息。例如:即使未安装CUDA工具包,410.72驱动程序也将显示CUDA VERSION 10.0

cuda安装

cuda下载地址
选择11.3进行下载
image.png
下载好后开始安装,按默认的安装目录安装即可
image.png
接下来按提示,选择了自定义安装,具体路径记录如下
image.png

pytorch安装

找了一个cuda11.3版本
image.png
在对应的环境下,这里是在yolov5下进行安装

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

安装好后打开python,测试是否可以用gpu,输入以下指令,输出True表示可以用gpu。
image.png

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐