【新手向】根据显卡型号安装合适的cuda和pytorch-Window11

查看显卡版本

此电脑-属性-设备管理器-显示适配器
【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第1张图片

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

nvidia-smi -L

image.png

nvidia驱动

到官网,根据上面查到的显卡信息,输入后获得合适的驱动程序,下载安装。
【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第2张图片

nvidia-smi

【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第3张图片
ps:与较新的驱动程序相关联的nvidia-smi显示的CUDA版本是DRIVER API兼容性版本。它根本不表示任何有关实际安装的CUDA版本的信息。例如:即使未安装CUDA工具包,410.72驱动程序也将显示CUDA VERSION 10.0

cuda安装

cuda下载地址
选择11.3进行下载
【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第4张图片
下载好后开始安装,按默认的安装目录安装即可
【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第5张图片
接下来按提示,选择了自定义安装,具体路径记录如下
【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第6张图片

pytorch安装

找了一个cuda11.3版本
【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第7张图片
在对应的环境下,这里是在yolov5下进行安装

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

安装好后打开python,测试是否可以用gpu,输入以下指令,输出True表示可以用gpu。
【新手向】根据显卡型号安装合适的cuda和pytorch-Window11_第8张图片

你可能感兴趣的:(pytorch,深度学习,人工智能)