3 cuda安装
3.1 查看自己计算机显卡所支持安装的cuda版本
1)从NVIDIA控制面板
获得显卡驱动型号和所支持下载的cuda最大版本号
打开NVIDIA控制面板,点击帮助里面的系统信息,在弹出的界面点击组件,NVCUDA64.DLL
所显示的产品名称,即为支持下载的cuda最高版本。
在弹出的界面点击显示,查看驱动程序版本。
2)Windows系统下cmd命令行获取显卡驱动版本和所支持下载的cuda最大版本号
打开cmd
,输入nvidia-smi
,显示如下内容:
上图中Driver Version
和CUDA Version
即为显卡驱动版本和所支持下载的cuda最大版本号。
3)结合查询CUDA发行文档说明,比照自己的显卡驱动版本和所支持下载的cuda最大版本号,确定自己能下载的CUDA版本。
网页地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
3.2 下载自己计算机所支持的CUDA版本
1)由于后续我需要安装pytorch1.8.1,所需cuda版本为11.1,所以这里我选择安装cuda11.1.1。
下载地址如下:
https://developer.nvidia.com/cuda-toolkit-archive