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