最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

Windows系统安装Pytorch 1.8.0 GPU记录

运维笔记admin10浏览0评论

**

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

发布评论

评论列表(0)

  1. 暂无评论