1 软件安装
1.1 安装Visual Studio
安装Visual Studio,社区版免费使用,使用社区版既可
1.2 安装Nvidia显卡驱动
安装Nvidia显卡驱动,安装显卡时注意显卡型号
NVIDIA驱动下载https://www.nvidia.cn/Download/index.aspx?lang=cn 安装完成之后可以在cmd中输入执行:
nvidia-smi
如果有错误会提示:
'nvidia-smi' 不是内部或外部命令,也不是可运行的程序 或批处理文件。
把C:\Program Files\NVIDIA Corporation\NVSMI添加到环境变量的path中。再重新打开cmd窗口。如果输出下图所示的显卡信息,说明你的驱动安装成功。
1.3 下载CUDA
CUDA安装版本注意与pytorch版本对应