1、nvidia-smi 查看显卡信息
nvidia-smi 指的是 NVIDIA System Management Interface;
在安装完成 NVIDIA 显卡驱动之后,对于 windows 用户而言,cmd 命令行界面还无法识别 nvidia-smi 命令,需要将相 关环境变量添加进去。如将 NVIDIA 显卡驱动安装在默认位置,nvidia-smi 命令所在1的完整路径应当为:
C:\Program Files\NVIDIA Corporation\NVSMI1
也即将上述路径添加进 Path 系统环境变量中。
2、查看 CUDA 信息
CUDA 的版本:
进入命令行:nvcc -V
3、查看显卡对应的cuda driver
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components
然后根据cuda版本选取适合的pytorch版本
4、到pytorch官网选择合适的版本
https://pytorch.org/get-started/locally/
下载即可