查看显卡版本
此电脑-属性-设备管理器-显示适配器
如果装了nvidia驱动的话也可以用命令列出所有GPU
nvidia-smi -L
nvidia驱动
到官网,根据上面查到的显卡信息,输入后获得合适的驱动程序,下载安装。
nvidia-smi
ps:与较新的驱动程序相关联的nvidia-smi显示的CUDA版本是DRIVER API兼容性版本。它根本不表示任何有关实际安装的CUDA版本的信息。例如:即使未安装CUDA工具包,410.72驱动程序也将显示CUDA VERSION 10.0
cuda安装
cuda下载地址
选择11.3进行下载
下载好后开始安装,按默认的安装目录安装即可
接下来按提示,选择了自定义安装,具体路径记录如下
pytorch安装
找了一个cuda11.3版本
在对应的环境下,这里是在yolov5下进行安装
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
安装好后打开python,测试是否可以用gpu,输入以下指令,输出True表示可以用gpu。