深度学习笔记:安装并使用pytorch的准备工作

查看电脑显卡

1.右击此电脑–设备管理器–显示设备器
深度学习笔记:安装并使用pytorch的准备工作_第1张图片

查看电脑驱动信息

打开cmd
依次输入以下代码,就可以看到显卡驱动的信息了。如图

cd C:\Program Files\NVIDIA Corporation\NVSMI
nvidia-smi

深度学习笔记:安装并使用pytorch的准备工作_第2张图片
如上图所示,当前驱动显卡版本为391.25
输入nvidia-smi -L命令查看显卡型号:
在这里插入图片描述

查询显卡驱动与CUDA对应表

驱动与CUDA对应表
深度学习笔记:安装并使用pytorch的准备工作_第3张图片
根据这个表中的对应关系安装,当没有版本对应的cuda时可去NVIDIA官网寻找对应显卡型号的驱动,进行驱动更新,如下图所示,搜索下载即可
深度学习笔记:安装并使用pytorch的准备工作_第4张图片
安装完成后,查看当前显卡版本,可以看到,显卡已经成功更新为517.48版本:
深度学习笔记:安装并使用pytorch的准备工作_第5张图片之后就可以开始下载需要的cuda了

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