Pytorch安装与基础知识

发布时间 2023-03-22 21:11:46作者: 杨谖之

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 中使用自己的环境进行安装。