**
Windows系统下的Pytorch 1.8.0 GPU版本安装步骤记录
**
在此记录自己安装CUDA、cuDNN、Pytorch 1.8.0以及tensorflow 1.14.0的整个过程。
一、CUDA的安装
首先确定了自己的电脑是可以安装CUDA的,之后确定本机的一些相关配置,包括驱动版本、CUDA版本来下载对应的CUDA toolkit。菜单栏点击进入NVIDIA Control Panel,左下角点击系统设置查看相应的驱动版本和CUDA版本如下图:
一般而言,不同版本的CUDA要求不同的NVIDIA驱动版本,同时驱动版本要不低于CUDA的安装版本,具体的对照关系如下:
官方链接如下:https://docs.nvidia/cuda/cuda-toolkit-release-notes/index.htm