admin 管理员组

文章数量: 1184232

**

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

本文标签： 系统安装 Windows pytorch GPU

版权声明：本文标题：Windows系统安装Pytorch 1.8.0 GPU记录 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1739729544a2083769.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。