admin管理员组

文章数量:1253712

**

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

本文标签: 系统安装WindowspytorchGPU