Pytorch安装与基础知识

Pytorch安装与基础知识

安装环境：Win10专业版
显卡：Nviida Geforce GTX 1660 Ti

安装 Anacodna

官网下载安装

安装 Cuda

Cuda 官网下载安装包。

进入 CMD，使用命令 nvcc -V 测试安装是否成功。

安装 cuDNN

虽然不知道为什么要安装，但是装就对了。

安装地址：cuDNN

根据安装的 Cuda 版本下载相应的 cuDNN。

解压下载的压缩包，获得三个文件夹，将三个文件夹复制到 Cuda 安装目录对应文件夹下，将安装目录的 include、lib、libnvvp 加入到系统环境变量的 Path 里。

如何测试 cuDNN 是否安装成功： Windows10检查Cuda和cuDNN是否安装成功？

安装 Pytorch

装了一圈，发现版本装错了，没办法，再装一个 11.8 版本的 Cuda，参考这篇文章：Win10系统下不同版本cuda切换。

安装完成之后，在 这个页面 获取安装命令，在 Anaconda 中使用自己的环境进行安装。